ObjectDetectionWOnnx

KexinFeng · Mar 15, 2023 · e352011 · e352011
1 parent ec79e87
commit e352011
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/src/main/java/ai/djl/examples/inference/ObjectDetection.java b/examples/src/main/java/ai/djl/examples/inference/ObjectDetection.java
@@ -66,7 +66,6 @@ public static DetectedObjects predict() throws IOException, ModelException, Tran
                         .optApplication(Application.CV.OBJECT_DETECTION)
                         .setTypes(Image.class, DetectedObjects.class)
                         .optFilter("backbone", backbone)
-                        .optEngine(Engine.getDefaultEngineName())
                         .optProgress(new ProgressBar())
                         .build();
 
@@ -86,7 +85,7 @@ private static void saveBoundingBoxImage(Image img, DetectedObjects detection)
 
         img.drawBoundingBoxes(detection);
 
-        Path imagePath = outputDir.resolve("detected-dog_bike_car.png");
+        Path imagePath = outputDir.resolve("dog_bike_car.png");
         // OpenJDK can't save jpg with alpha channel
         img.save(Files.newOutputStream(imagePath), "png");
         logger.info("Detected objects image has been saved in: {}", imagePath);