Преглед на файлове

v4.14.0 增加ai识别逻辑

dengsixing преди 5 месеца
родител
ревизия
743b2a1630

+ 5 - 0
src/main/java/com/fdkankan/contro/mq/service/impl/BuildSceneServiceImpl.java

@@ -136,6 +136,8 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
     private ISceneColdStorageService sceneColdStorageService;
     @Autowired
     private IWbService wbService;
+    @Autowired
+    private IAiService aiService;
 
 
     @Override
@@ -372,6 +374,9 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
             String pushToken = fdageData.getString("pushToken");
             this.pushMsgToApp(pushChannel,pushToken, cameraType, scenePlus.getTitle(), scenePlusExt.getWebSite());
 
+            //ai识别
+            aiService.detectScenePano(sceneCode, path);
+
 //            //删除计算目录
             if(CollUtil.isEmpty(notDeleteNasNumList) || !notDeleteNasNumList.contains(sceneCode)){
                 CreateObjUtil.deleteFile(path.replace(ConstantFilePath.BUILD_MODEL_PATH, "/"));

+ 2 - 0
src/main/java/com/fdkankan/contro/service/ISceneMarkShapeService.java

@@ -10,4 +10,6 @@ import com.fdkankan.contro.entity.SceneMarkShape;
 public interface ISceneMarkShapeService extends IService<SceneMarkShape> {
     SceneMarkShape findByNumAndImagePathAndType(String num, String imagePath,Integer type);
 
+    SceneMarkShape findByNumAndType(String num,Integer type);
+
 }

+ 5 - 0
src/main/java/com/fdkankan/contro/service/impl/AiServiceImpl.java

@@ -83,6 +83,11 @@ public class AiServiceImpl implements IAiService {
                 log.error("ai识别报错,inPath:{}", absolutePath, e);
             }
         }
+
+        //生成ai.json
+        SceneMarkShape sceneMarkShapes = sceneMarkShapeService.findByNumAndType(num, DetectType.PANO.getCode());
+
+
     }
 
     @Override

+ 8 - 0
src/main/java/com/fdkankan/contro/service/impl/SceneMarkShapeServiceImpl.java

@@ -31,5 +31,13 @@ public class SceneMarkShapeServiceImpl extends ServiceImpl<MarkShapeMapper, Scen
         return getOne(wrapper);
     }
 
+    @Override
+    public SceneMarkShape findByNumAndType(String num, Integer type) {
+        LambdaQueryWrapper<SceneMarkShape> wrapper = Wrappers.lambdaQuery();
+        wrapper.eq(SceneMarkShape::getNum,num);
+        wrapper.eq(SceneMarkShape::getType,type);
+        return getOne(wrapper);
+    }
+
 
 }