diff --git a/examples/vision-example/pom.xml b/examples/vision-example/pom.xml
index ebca5de69f6a444030cdc24e27511d60430f0f25..79f6a4d27abcdbe2cec0a59c31f7f3df75b2e78f 100644
--- a/examples/vision-example/pom.xml
+++ b/examples/vision-example/pom.xml
@@ -136,7 +136,7 @@
ai.djl.tensorflow
tensorflow-native-cpu
- win-x86_64
+ ${djl.platform.windows-x86_64}
runtime
2.16.1
@@ -144,7 +144,7 @@
ai.djl.mxnet
mxnet-native-mkl
- win-x86_64
+ ${djl.platform.windows-x86_64}
runtime
1.9.1
@@ -179,6 +179,7 @@
${javacv.platform.linux-x86_64}
+
ai.djl.pytorch
pytorch-native-cpu
@@ -190,14 +191,14 @@
ai.djl.tensorflow
tensorflow-native-cpu
- linux-x86_64
+ ${djl.platform.linux-x86_64}
runtime
2.16.1
ai.djl.mxnet
mxnet-native-mkl
- linux-x86_64
+ ${djl.platform.linux-x86_64}
runtime
1.9.1
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 d1d5a638159781a22f27883af51dea66c738eea5..923252defd8b7e237d33f95d89e979bdb65d492b 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 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 bebe389c4ebc1d7ff2748a1b5bcf5258be4328d2..1101868cc546fed2e195b4ae6154a557651a1314 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 5b28193721d5c7305f77af61dd3cb19ac1661d8d..a786085490f8f78960fb1900c023787fcdcec8fb 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);
}