Bladeren bron

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

xiewj 2 jaren geleden
bovenliggende
commit
d6a8c0d29c
1 gewijzigde bestanden met toevoegingen van 5 en 1 verwijderingen
  1. 5 1
      src/main/java/com/fdkankan/scene/controller/SceneMarkShapeController.java

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