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); }