From d63e36cd99b38c8e44d89f181c270658c1b5d88d Mon Sep 17 00:00:00 2001
From: deng_wu_jie <410039586@qq.com>
Date: Thu, 11 Sep 2025 09:52:24 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96vision=20demo?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 examples/vision-example/pom.xml                          | 9 +++++----
 .../java/smartai/examples/vision/InstanceSegDemo.java    | 2 +-
 .../main/java/smartai/examples/vision/PoseDetDemo.java   | 6 ++----
 .../main/java/smartai/examples/vision/SemSegDemo.java    | 1 -
 4 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/examples/vision-example/pom.xml b/examples/vision-example/pom.xml
index ebca5de..79f6a4d 100644
--- a/examples/vision-example/pom.xml
+++ b/examples/vision-example/pom.xml
@@ -136,7 +136,7 @@
         <dependency>
             <groupId>ai.djl.tensorflow</groupId>
             <artifactId>tensorflow-native-cpu</artifactId>
-            <classifier>win-x86_64</classifier>
+            <classifier>${djl.platform.windows-x86_64}</classifier>
             <scope>runtime</scope>
             <version>2.16.1</version>
         </dependency>
@@ -144,7 +144,7 @@
         <dependency>
             <groupId>ai.djl.mxnet</groupId>
             <artifactId>mxnet-native-mkl</artifactId>
-            <classifier>win-x86_64</classifier>
+            <classifier>${djl.platform.windows-x86_64}</classifier>
             <scope>runtime</scope>
             <version>1.9.1</version>
         </dependency>
@@ -179,6 +179,7 @@
             <classifier>${javacv.platform.linux-x86_64}</classifier>
         </dependency>
 
+        <!--PyTorch离线平台依赖-->
         <dependency>
             <groupId>ai.djl.pytorch</groupId>
             <artifactId>pytorch-native-cpu</artifactId>
@@ -190,14 +191,14 @@
         <dependency>
             <groupId>ai.djl.tensorflow</groupId>
             <artifactId>tensorflow-native-cpu</artifactId>
-            <classifier>linux-x86_64</classifier>
+            <classifier>${djl.platform.linux-x86_64}</classifier>
             <scope>runtime</scope>
             <version>2.16.1</version>
         </dependency>
         <dependency>
             <groupId>ai.djl.mxnet</groupId>
             <artifactId>mxnet-native-mkl</artifactId>
-            <classifier>linux-x86_64</classifier>
+            <classifier>${djl.platform.linux-x86_64}</classifier>
             <scope>runtime</scope>
             <version>1.9.1</version>
         </dependency>
diff --git a/examples/vision-example/src/main/java/smartai/examples/vision/InstanceSegDemo.java b/examples/vision-example/src/main/java/smartai/examples/vision/InstanceSegDemo.java
index d1d5a63..923252d 100644
--- a/examples/vision-example/src/main/java/smartai/examples/vision/InstanceSegDemo.java
+++ b/examples/vision-example/src/main/java/smartai/examples/vision/InstanceSegDemo.java
@@ -78,7 +78,7 @@ public class InstanceSegDemo {
             Image image = SmartImageFactory.getInstance().fromFile(Paths.get("src/main/resources/dog_bike_car.jpg"));
             R<DetectionResponse> result = detectorModel.detect(image);
             if(result.isSuccess()){
-                log.info("实例分割结果：{}", result.getData());
+                log.info("实例分割结果：{}", JSONObject.toJSONString(result.getData()));
             }else{
                 log.info("实例分割失败：{}", result.getMessage());
             }
diff --git a/examples/vision-example/src/main/java/smartai/examples/vision/PoseDetDemo.java b/examples/vision-example/src/main/java/smartai/examples/vision/PoseDetDemo.java
index bebe389..1101868 100644
--- a/examples/vision-example/src/main/java/smartai/examples/vision/PoseDetDemo.java
+++ b/examples/vision-example/src/main/java/smartai/examples/vision/PoseDetDemo.java
@@ -41,14 +41,12 @@ public class PoseDetDemo {
      * 获取姿态估计模型
      * 注意事项：
      * 1、更多模型请查看文档：http://doc.smartjavaai.cn
-     * 2、模型可检测物体请查看：模型同目录文件synset.txt
      */
     public PoseModel getModel(){
         PoseModelConfig config = new PoseModelConfig();
         //姿态估计模型，切换模型需要同时修改modelEnum及modelPath
-        config.setModelEnum(PoseModelEnum.YOLOV8N_POSE_PT);
-        //模型所在路径，synset.txt也需要放在同目录下
-        config.setModelPath("/Users/wenjie/Documents/develop/model/vision/pose/yolo11n-pose-onnx/yolo11n-pose.onnx");
+        config.setModelEnum(PoseModelEnum.YOLO11N_POSE_PT);
+        config.setModelPath("/Users/wenjie/Documents/develop/model/vision/pose/yolo11n-pose-pt");
         config.setDevice(device);
         //置信度阈值
         config.setThreshold(0.25f);
diff --git a/examples/vision-example/src/main/java/smartai/examples/vision/SemSegDemo.java b/examples/vision-example/src/main/java/smartai/examples/vision/SemSegDemo.java
index 5b28193..a786085 100644
--- a/examples/vision-example/src/main/java/smartai/examples/vision/SemSegDemo.java
+++ b/examples/vision-example/src/main/java/smartai/examples/vision/SemSegDemo.java
@@ -53,7 +53,6 @@ public class SemSegDemo {
         config.setModelPath("/Users/wenjie/Documents/develop/model/vision/semseg/deeplabv3/deeplabv3.pt");
         // 指定允许的类别
 //            config.setAllowedClasses(Arrays.asList("person","car"));
-        //指定返回检测数量
         config.setDevice(device);
         return SemSegModelFactory.getInstance().getModel(config);
     }
-- 
Gitee