From 7b3b249a7f2320f97e21e94e26a65f4b4ead0b6e Mon Sep 17 00:00:00 2001
From: shikeying <shikeying@163.com>
Date: 星期六, 24 九月 2022 16:50:25 +0800
Subject: [PATCH] 视频相似度分析1

---
 recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 deletions(-)

diff --git a/recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java b/recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java
index ef88d0e..d1e3849 100644
--- a/recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java
+++ b/recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java
@@ -1,13 +1,35 @@
 package com.iplatform.recvideo;
 
+import com.iplatform.model.po.Rc_video_t1;
+import com.iplatform.recvideo.util.PythonInvokeUtils;
 import com.iplatform.recvideo.util.VideoFileUtils;
 import org.junit.Test;
+import org.springframework.web.client.RestTemplate;
 
 import java.util.List;
 
 public class VideoSimilarTest {
 
     @Test
+    public void testPythonSearch() throws Exception{
+        String url = "http://121.36.40.27:12345/ai/video/search_img";
+        RestTemplate restTemplate = new RestTemplate();
+        List<Rc_video_t1> data = PythonInvokeUtils.acquireImageSearchResult("landscape_01"
+                , "/opt/ai/video/20220921/landscape_01/landscape_01_1.jpg", "30", url, restTemplate);
+        if(data != null){
+            for(Rc_video_t1 e : data){
+                System.out.println(e);
+            }
+        }
+    }
+
+//    @Test
+    public void testGetVideoId(){
+        String id = PythonInvokeUtils.getFileNameWithoutSuffix("/opt/ai/video/20220921/landscape_01.mp4", Constants.VIDEO_SUFFIX);
+        System.out.println(id);
+    }
+
+//    @Test
     public void testVideoFolderInfo(){
         List<VideoFolderInfo> list = VideoFileUtils.getBatchVideoFolderInfo("D:/dev_tools/ai", "video");
         if(list != null){

--
Gitblit v1.9.1