- 【人脸检测】新增6个模型(MTCNN、YOLOV5、RetinaFace小尺寸版)，大幅提升性能

- 【人脸识别】新增Seetaface6轻量模型 - 【目标检测】支持视频流目标检测（rtsp、视频文件等） - 【目标检测】支持tensorflow2目标检测模型 - 【目标检测】新增行人检测模型(yolo-person) - 【通用视觉】新增4个动作识别模型 - 【通用视觉】新增语义分割模型 - 【通用视觉】新增5个实例分割模型(含yolov8-seg、yolov11-seg) - 【通用视觉】新增yolo-obb11旋转框检测(含yolov11-obb) - 【通用视觉】新增5个姿态估计模型(含yolov8-pose、yolov11-pose)
2026-09-10 03:28:49 +00:00 · 2025-09-07 17:19:19 +08:00
parent 2b044fda29
commit a8e7ce6c4e
102 changed files with 4473 additions and 1368 deletions
--- a/examples/ocr-examples/output/plate_recognized2.jpg
+++ b/examples/ocr-examples/output/plate_recognized2.jpg
--- a/examples/ocr-examples/pom.xml
+++ b/examples/ocr-examples/pom.xml
@@ -34,7 +34,7 @@
        <dependencies>
            <dependency>
                <groupId>cn.smartjavaai</groupId>
-                <artifactId>smartjavaai-bom</artifactId>
+                <artifactId>bom</artifactId>
                <version>${smartjavaai.version}</version>
                <type>pom</type>
                <!-- 注意这里是import -->
@@ -50,16 +50,13 @@
            <artifactId>commons-cli</artifactId>
            <version>1.9.0</version>
        </dependency>
+
        <dependency>
            <groupId>commons-io</groupId>
            <artifactId>commons-io</artifactId>
            <version>2.17.0</version>
        </dependency>
-        <dependency>
-            <groupId>org.apache.logging.log4j</groupId>
-            <artifactId>log4j-slf4j2-impl</artifactId>
-            <version>2.24.1</version>
-        </dependency>
+
        <dependency>
            <groupId>org.testng</groupId>
            <artifactId>testng</artifactId>
@@ -73,11 +70,6 @@
            <artifactId>logback-classic</artifactId>
            <version>1.2.3</version>
        </dependency>
-        <dependency>
-            <groupId>org.slf4j</groupId>
-            <artifactId>slf4j-api</artifactId>
-            <version>1.7.30</version>
-        </dependency>

        <dependency>
            <groupId>com.alibaba</groupId>
@@ -96,14 +88,14 @@
        <!--OCR检测模块-->
        <dependency>
            <groupId>cn.smartjavaai</groupId>
-            <artifactId>smartjavaai-ocr</artifactId>
+            <artifactId>ocr</artifactId>
        </dependency>


        <dependency>
            <groupId>ai.djl.pytorch</groupId>
            <artifactId>pytorch-jni</artifactId>
-            <version>2.7.1-0.34.0</version>
+            <version>2.5.1-0.32.0</version>
            <scope>runtime</scope>
        </dependency>

@@ -140,7 +132,7 @@
            <groupId>ai.djl.pytorch</groupId>
            <artifactId>pytorch-native-cpu</artifactId>
            <classifier>${djl.platform.windows-x86_64}</classifier>
-            <version>2.7.1</version>
+            <version>2.5.1</version>
            <scope>runtime</scope>
        </dependency>

@@ -178,7 +170,14 @@
            <groupId>ai.djl.pytorch</groupId>
            <artifactId>pytorch-native-cpu</artifactId>
            <classifier>${djl.platform.linux-x86_64}</classifier>
-            <version>2.7.1</version>
+            <version>2.5.1</version>
+            <scope>runtime</scope>
+        </dependency>
+        <dependency>
+            <groupId>ai.djl.pytorch</groupId>
+            <artifactId>pytorch-native-cpu-precxx11</artifactId>
+            <classifier>${djl.platform.linux-x86_64}</classifier>
+            <version>2.5.1</version>
            <scope>runtime</scope>
        </dependency>

@@ -215,7 +214,7 @@
            <groupId>ai.djl.pytorch</groupId>
            <artifactId>pytorch-native-cpu</artifactId>
            <classifier>${djl.platform.osx-aarch64}</classifier>
-            <version>2.7.1</version>
+            <version>2.5.1</version>
            <scope>runtime</scope>
        </dependency>

@@ -249,13 +248,7 @@
            <classifier>${javacv.platform.linux-arm64}</classifier>
        </dependency>

-        <dependency>
-            <groupId>ai.djl.pytorch</groupId>
-            <artifactId>pytorch-native-cpu-precxx11</artifactId>
-            <classifier>${djl.platform.linux-aarch64}</classifier>
-            <version>2.7.1</version>
-            <scope>runtime</scope>
-        </dependency>
+



--- a/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrDetectionDemo.java
+++ b/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrDetectionDemo.java
@@ -29,6 +29,7 @@ import java.util.List;
 /**
 * OCR 文本检测 示例
 * 模型下载地址：https://pan.baidu.com/s/15Noz2xHQzqMQSl1B19BobQ?pwd=1234 提取码: 1234
+ * 开发文档：http://doc.smartjavaai.cn/
 * @author dwj
 */
@Slf4j
@@ -52,7 +53,7 @@ public class OcrDetectionDemo {
     */
    public OcrCommonDetModel getDetectionModel() {
        OcrDetModelConfig config = new OcrDetModelConfig();
-        //指定检测模型
+        //指定检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(CommonDetModelEnum.PP_OCR_V5_MOBILE_DET_MODEL);
        //指定模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        config.setDetModelPath("/Users/xxx/Documents/develop/model/ocr/PP-OCRv5_mobile_det_infer/PP-OCRv5_mobile_det_infer.onnx");
--- a/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrDirectionDetDemo.java
+++ b/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrDirectionDetDemo.java
@@ -22,8 +22,8 @@ import java.util.List;
 /**
 * OCR 行文本方向检测 示例
 * 模型下载地址：https://pan.baidu.com/s/1MLfd73Vjdpnuls9-oqc9uw?pwd=1234 提取码: 1234
+ * 开发文档：http://doc.smartjavaai.cn/
 * @author dwj
- * @date 2025/5/25
 */
@Slf4j
 public class OcrDirectionDetDemo {
@@ -43,7 +43,7 @@ public class OcrDirectionDetDemo {
     */
    public OcrDirectionModel getDirectionModel(){
        DirectionModelConfig directionModelConfig = new DirectionModelConfig();
-        //指定行文本方向检测模型
+        //指定行文本方向检测模型，切换模型需要同时修改modelEnum及modelPath
        directionModelConfig.setModelEnum(DirectionModelEnum.PP_LCNET_X0_25);
        //指定行文本方向检测模型路径，需要更改为自己的模型路径（下载地址请查看文档）
        directionModelConfig.setModelPath("/Users/xxx/Documents/develop/model/ocr/PP-LCNet_x0_25_textline_ori_infer/PP-LCNet_x0_25_textline_ori_infer.onnx");
@@ -58,7 +58,7 @@ public class OcrDirectionDetDemo {
     */
    public OcrCommonDetModel getDetectionModel() {
        OcrDetModelConfig config = new OcrDetModelConfig();
-        //指定检测模型
+        //指定检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(CommonDetModelEnum.PP_OCR_V5_MOBILE_DET_MODEL);
        //指定模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        config.setDetModelPath("/Users/xxx/Documents/develop/model/ocr/PP-OCRv5_mobile_det_infer/PP-OCRv5_mobile_det_infer.onnx");
--- a/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrRecognizeDemo.java
+++ b/examples/ocr-examples/src/main/java/smartai/examples/ocr/common/OcrRecognizeDemo.java
@@ -36,8 +36,8 @@ import java.util.List;
 /**
 * OCR 文本识别 示例
 * 模型下载地址：https://pan.baidu.com/s/1MLfd73Vjdpnuls9-oqc9uw?pwd=1234 提取码: 1234
+ * 开发文档：http://doc.smartjavaai.cn/
 * @author dwj
- * @date 2025/5/25
 */
@Slf4j
 public class OcrRecognizeDemo {
@@ -58,7 +58,7 @@ public class OcrRecognizeDemo {
     */
    public OcrCommonRecModel getRecModel(){
        OcrRecModelConfig recModelConfig = new OcrRecModelConfig();
-        //指定文本识别模型
+        //指定文本识别模型，切换模型需要同时修改modelEnum及modelPath
        recModelConfig.setRecModelEnum(CommonRecModelEnum.PP_OCR_V5_MOBILE_REC_MODEL);
        //指定识别模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        recModelConfig.setRecModelPath("/Users/wenjie/Documents/develop/model/ocr/PP-OCRv5_server_rec_infer/PP-OCRv5_server_rec.onnx");
@@ -73,7 +73,7 @@ public class OcrRecognizeDemo {
     */
    public OcrCommonDetModel getDetectionModel() {
        OcrDetModelConfig config = new OcrDetModelConfig();
-        //指定检测模型
+        //指定检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(CommonDetModelEnum.PP_OCR_V5_MOBILE_DET_MODEL);
        //指定模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        config.setDetModelPath("/Users/wenjie/Documents/develop/model/ocr/PP-OCRv5_server_det_infer/PP-OCRv5_server_det.onnx");
@@ -87,7 +87,7 @@ public class OcrRecognizeDemo {
     */
    public OcrDirectionModel getDirectionModel(){
        DirectionModelConfig directionModelConfig = new DirectionModelConfig();
-        //指定行文本方向检测模型
+        //指定行文本方向检测模型，切换模型需要同时修改modelEnum及modelPath
        directionModelConfig.setModelEnum(DirectionModelEnum.PP_LCNET_X0_25);
        //指定行文本方向检测模型路径，需要更改为自己的模型路径（下载地址请查看文档）
        directionModelConfig.setModelPath("/Users/xxx/Documents/develop/model/ocr/PP-LCNet_x0_25_textline_ori_infer/PP-LCNet_x0_25_textline_ori_infer.onnx");
@@ -102,7 +102,7 @@ public class OcrRecognizeDemo {
     */
    public OcrCommonRecModel getRecModelWithDirection() {
        OcrRecModelConfig recModelConfig = new OcrRecModelConfig();
-        //指定文本识别模型
+        //指定文本识别模型，切换模型需要同时修改modelEnum及modelPath
        recModelConfig.setRecModelEnum(CommonRecModelEnum.PP_OCR_V5_MOBILE_REC_MODEL);
        //指定识别模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        recModelConfig.setRecModelPath("/Users/xxx/Documents/develop/model/ocr/PP-OCRv5_mobile_rec_infer/PP-OCRv5_mobile_rec_infer.onnx");
--- a/examples/ocr-examples/src/main/java/smartai/examples/ocr/plate/PlateRecDemo.java
+++ b/examples/ocr-examples/src/main/java/smartai/examples/ocr/plate/PlateRecDemo.java
@@ -17,11 +17,17 @@ import lombok.extern.slf4j.Slf4j;
 import org.junit.BeforeClass;
 import org.junit.Test;

+import javax.imageio.ImageIO;
+import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
+import java.nio.file.Paths;
 import java.util.List;

 /**
+ * 车牌识别demo
+ * 模型下载地址：https://pan.baidu.com/s/1YEP56UqYcL-Op80M6JAreA?pwd=1234 提取码: 1234
+ * 开发文档：http://doc.smartjavaai.cn/
 * @author dwj
 */
@Slf4j
@@ -42,8 +48,10 @@ public class PlateRecDemo {
     */
    public PlateDetModel getPlateDetModel() {
        PlateDetModelConfig config = new PlateDetModelConfig();
+        //车牌检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(PlateDetModelEnum.YOLOV5);
-        config.setModelPath("/Users/xxx/Documents/develop/model/plate/yolov5_plate_detect.onnx");
+        //下载模型并替换本地路径
+        config.setModelPath("/Users/wenjie/Documents/develop/model/plate/yolov5_plate_detect.onnx");
        config.setDevice(device);
        return PlateModelFactory.getInstance().getDetModel(config);
    }
@@ -54,15 +62,22 @@ public class PlateRecDemo {
     */
    public PlateRecModel getPlateRecModel() {
        PlateRecModelConfig recModelConfig = new PlateRecModelConfig();
+        //车牌识别模型，切换模型需要同时修改modelEnum及modelPath
        recModelConfig.setModelEnum(PlateRecModelEnum.PLATE_REC_CRNN);
-        recModelConfig.setModelPath("/Users/xxx/Documents/develop/model/plate/plate_rec_color.onnx");
+        //下载模型并替换本地路径
+        recModelConfig.setModelPath("/Users/wenjie/Documents/develop/model/plate/plate_rec_color.onnx");
+        //指定车牌检测模型
        recModelConfig.setPlateDetModel(getPlateDetModel());
        return PlateModelFactory.getInstance().getRecModel(recModelConfig);
    }

+    /**
+     * 车牌识别
+     */
    @Test
    public void testDetect() {
        PlateRecModel plateRecModel = getPlateRecModel();
+        //识别车号
        R<List<PlateInfo>> result = plateRecModel.recognize("src/main/resources/plate/Quicker_20220930_180856.png");
        if(result.isSuccess()){
            log.info("车牌识别结果：{}", JsonUtils.toJson(result.getData()));
@@ -71,9 +86,13 @@ public class PlateRecDemo {
        }
    }

+    /**
+     * 车牌识别及绘制结果
+     */
    @Test
    public void recognizeAndDraw() {
        PlateRecModel plateRecModel = getPlateRecModel();
+        //识别车号并绘制结果
        R<Void> result = plateRecModel.recognizeAndDraw("src/main/resources/plate/single_green.jpg", "output/plate_recognized2.jpg");
        if(result.isSuccess()){
            log.info("车牌识别成功");
@@ -83,5 +102,28 @@ public class PlateRecDemo {

    }

+    /**
+     * 车牌识别及绘制结果
+     */
+    @Test
+    public void recognizeAndDraw2() {
+        try {
+            PlateRecModel plateRecModel = getPlateRecModel();
+            BufferedImage image = null;
+            String imagePath = "src/main/resources/plate/Quicker_20220930_180856.png";
+            image = ImageIO.read(new File(Paths.get(imagePath).toAbsolutePath().toString()));
+            //可以根据后续业务场景使用detectedImage
+            R<BufferedImage> detectedImage = plateRecModel.recognizeAndDraw(image);
+            if(detectedImage.isSuccess()){
+                log.info("车牌识别成功");
+                ImageUtils.saveImage(detectedImage.getData(), "output/plate_recognized2.jpg");
+            }else{
+                log.error("车牌识别失败：{}", detectedImage.getMessage());
+            }
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+

 }
--- a/examples/ocr-examples/src/main/java/smartai/examples/ocr/table/TableRecDemo.java
+++ b/examples/ocr-examples/src/main/java/smartai/examples/ocr/table/TableRecDemo.java
@@ -38,8 +38,8 @@ import java.util.List;
 /**
 * OCR 表格识别 示例
 * 模型下载地址：https://pan.baidu.com/s/1MLfd73Vjdpnuls9-oqc9uw?pwd=1234 提取码: 1234
+ * 开发文档：http://doc.smartjavaai.cn/
 * @author dwj
- * @date 2025/5/25
 */
@Slf4j
 public class TableRecDemo {
@@ -60,7 +60,7 @@ public class TableRecDemo {
     */
    public OcrCommonRecModel getRecModel(){
        OcrRecModelConfig recModelConfig = new OcrRecModelConfig();
-        //指定文本识别模型
+        //指定文本识别模型，切换模型需要同时修改modelEnum及modelPath
        recModelConfig.setRecModelEnum(CommonRecModelEnum.PP_OCR_V5_MOBILE_REC_MODEL);
        //指定识别模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        recModelConfig.setRecModelPath("/Users/wenjie/Documents/develop/model/ocr/PP-OCRv5_mobile_rec_infer/PP-OCRv5_mobile_rec_infer.onnx");
@@ -75,7 +75,7 @@ public class TableRecDemo {
     */
    public OcrCommonDetModel getDetectionModel() {
        OcrDetModelConfig config = new OcrDetModelConfig();
-        //指定检测模型
+        //指定检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(CommonDetModelEnum.PP_OCR_V5_MOBILE_DET_MODEL);
        //指定模型位置，需要更改为自己的模型路径（下载地址请查看文档）
        config.setDetModelPath("/Users/wenjie/Documents/develop/model/ocr/PP-OCRv5_mobile_det_infer/PP-OCRv5_mobile_det_infer.onnx");
@@ -90,7 +90,7 @@ public class TableRecDemo {
     */
    public OcrDirectionModel getDirectionModel(){
        DirectionModelConfig directionModelConfig = new DirectionModelConfig();
-        //指定行文本方向检测模型
+        //指定行文本方向检测模型，切换模型需要同时修改modelEnum及modelPath
        directionModelConfig.setModelEnum(DirectionModelEnum.PP_LCNET_X0_25);
        //指定行文本方向检测模型路径，需要更改为自己的模型路径（下载地址请查看文档）
        directionModelConfig.setModelPath("/Users/wenjie/Documents/develop/model/ocr/PP-LCNet_x0_25_textline_ori_infer/PP-LCNet_x0_25_textline_ori_infer.onnx");
@@ -104,7 +104,7 @@ public class TableRecDemo {
     */
    public TableStructureModel getTableStructureModel(){
        TableStructureConfig config = new TableStructureConfig();
-        //指定行文本方向检测模型
+        //指定行文本方向检测模型，切换模型需要同时修改modelEnum及modelPath
        config.setModelEnum(TableStructureModelEnum.SLANET_PLUS);
        //指定行文本方向检测模型路径，需要更改为自己的模型路径（下载地址请查看文档）
        config.setModelPath("/Users/wenjie/Documents/develop/model/ocr/slanet-plus/slanet-plus.onnx");