Fix bug in OD TF saved model example (#2050)

deepjavalibrary · Sep 29, 2022 · d306d2f · d306d2f
1 parent 53022b9
commit d306d2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ples/src/main/java/ai/djl/examples/inference/ObjectDetectionWithTensorflowSavedModel.java b/...ples/src/main/java/ai/djl/examples/inference/ObjectDetectionWithTensorflowSavedModel.java
@@ -88,7 +88,7 @@ public static DetectedObjects predict() throws IOException, ModelException, Tran
                         .setTypes(Image.class, DetectedObjects.class)
                         .optModelUrls(modelUrl)
                         // saved_model.pb file is in the subfolder of the model archive file
-                        .optModelName("ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model")
+                        .optModelName("saved_model")
                         .optTranslator(new MyTranslator())
                         .optEngine("TensorFlow")
                         .optProgress(new ProgressBar())