From a78b76b664830a1ac691396d5cb64166c6919ff1 Mon Sep 17 00:00:00 2001 From: shikeying <shikeying@163.com> Date: 星期五, 30 九月 2022 18:12:44 +0800 Subject: [PATCH] 视频相似度分析5 --- recommend-video/src/test/java/com/iplatform/recvideo/VideoSimilarTest.java | 35 ++++++++++++++++++++++++++++++++++- 1 files changed, 34 insertions(+), 1 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..3c7e781 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,46 @@ 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 com.walker.infrastructure.utils.ClassUtils; +import com.walker.scheduler.impl.TimedScheduler; import org.junit.Test; +import org.springframework.web.client.RestTemplate; +import java.lang.reflect.Constructor; import java.util.List; public class VideoSimilarTest { - @Test +// @Test + public void testSchedulerClass() throws Exception{ + Class<?> schedulerClass = ClassUtils.forName("com.iplatform.recvideo.scheduler.VideoSearchScheduler", VideoSimilarTest.class.getClassLoader()); + Constructor<TimedScheduler> constructor = (Constructor<TimedScheduler>)schedulerClass.getConstructor(int.class, String.class); + TimedScheduler listeningScheduler = constructor.newInstance(100, "test_scheduler"); + + } + +// @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