【人脸识别】 新增多种人脸识别模型

【底层优化】 支持自由选择 OpenCV 或 BufferedImage 作为图像引擎

【通用图像】 全部模型启用 Image 输入，支持各类图片格式与 Image 的互转

【模型管理】 优化模型生命周期，关闭后可重新创建

【人脸识别】 支持在人脸查询结果中绘制姓名标注

【人脸检测】 新增人脸裁剪功能

【修复】 修复若干已知问题，提升系统稳定性

vision/src/main/java/cn/smartjavaai/action/model/ActionRecModel.java

@@ -34,4 +34,8 @@ public interface ActionRecModel extends AutoCloseable{
     }
 
 
+    default void setFromFactory(boolean fromFactory){
+        throw new UnsupportedOperationException("默认不支持该功能");
+    }
+
 }

vision/src/main/java/cn/smartjavaai/action/model/ActionRecModelFactory.java

@@ -76,6 +76,7 @@ public class ActionRecModelFactory {
             throw new DetectionException(e);
         }
         model.loadModel(config);
+        model.setFromFactory(true);
         return model;
     }
 
@@ -90,14 +91,6 @@ public class ActionRecModelFactory {
     }
 
 
-    /**
-     * 移除缓存的模型
-     * @param modelEnum
-     */
-    public static void removeFromCache(ActionRecModelEnum modelEnum) {
-        modelMap.remove(modelEnum);
-    }
-
 
     // 初始化默认算法
     static {
@@ -107,5 +100,27 @@ public class ActionRecModelFactory {
         registerAlgorithm(ActionRecModelEnum.VIT_BASE_PATCH16_224_DJL, CommonActionRecModel.class);
         log.debug("缓存目录：{}", Config.getCachePath());
     }
+
+    /**
+     * 关闭所有已加载的模型
+     */
+    public void closeAll() {
+        modelMap.values().forEach(model -> {
+            try {
+                model.close();
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        });
+        modelMap.clear();
+    }
+
+    /**
+     * 移除缓存的模型
+     * @param modelEnum
+     */
+    public static void removeFromCache(ActionRecModelEnum modelEnum) {
+        modelMap.remove(modelEnum);
+    }
 }
 

vision/src/main/java/cn/smartjavaai/action/model/CommonActionRecModel.java

@@ -110,6 +110,9 @@ public class CommonActionRecModel implements ActionRecModel{
 
     @Override
     public void close() throws Exception {
+        if (fromFactory) {
+            ActionRecModelFactory.removeFromCache(config.getModelEnum());
+        }
         try {
             if (predictorPool != null) {
                 predictorPool.close();
@@ -125,4 +128,14 @@ public class CommonActionRecModel implements ActionRecModel{
             log.warn("关闭 model 失败", e);
         }
     }
+
+    private boolean fromFactory = false;
+
+    @Override
+    public void setFromFactory(boolean fromFactory) {
+        this.fromFactory = fromFactory;
+    }
+    public boolean isFromFactory() {
+        return fromFactory;
+    }
 }