Browse Source

增加全景图识别推理队列参数

xiewj 2 years ago
parent
commit
d6a8c0d29c

+ 5 - 1
src/main/java/com/fdkankan/scene/controller/SceneMarkShapeController.java

@@ -13,6 +13,7 @@ import com.fdkankan.scene.service.ISceneMarkShapeService;
 import com.fdkankan.scene.service.IScenePlusExtService;
 import com.fdkankan.scene.service.IScenePlusExtService;
 import com.fdkankan.scene.service.IScenePlusService;
 import com.fdkankan.scene.service.IScenePlusService;
 import com.fdkankan.scene.vo.FileParamVO;
 import com.fdkankan.scene.vo.FileParamVO;
+import com.fdkankan.scene.vo.SceneMarkShapeDetectParamVO;
 import com.fdkankan.scene.vo.SceneMarkShapeParamVO;
 import com.fdkankan.scene.vo.SceneMarkShapeParamVO;
 import com.fdkankan.web.controller.BaseController;
 import com.fdkankan.web.controller.BaseController;
 import com.fdkankan.web.response.Result;
 import com.fdkankan.web.response.Result;
@@ -54,6 +55,8 @@ public class SceneMarkShapeController extends BaseController
     private String ossUrlPrefix;
     private String ossUrlPrefix;
     @Autowired
     @Autowired
     private RabbitMqProducer rabbitMqProducer;
     private RabbitMqProducer rabbitMqProducer;
+    @Value("${main.url}")
+    private String mainUrl;
     /**
     /**
      * 获取场景全景图路径连接
      * 获取场景全景图路径连接
      */
      */
@@ -83,7 +86,8 @@ public class SceneMarkShapeController extends BaseController
      * 将需要推理的场景推入到队列
      * 将需要推理的场景推入到队列
      */
      */
     @PostMapping("/detectScene")
     @PostMapping("/detectScene")
-    public ResultData detect(@RequestBody @Validated  SceneMarkShapeParamVO param) {
+    public ResultData detect(@RequestBody @Validated SceneMarkShapeDetectParamVO param) {
+        param.setWebSite(mainUrl);
         rabbitMqProducer.sendByWorkQueue(yolov5DetectQueue, param);
         rabbitMqProducer.sendByWorkQueue(yolov5DetectQueue, param);
         return ResultData.ok();
         return ResultData.ok();
     }
     }