|
@@ -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();
|
|
|
}
|