From 3c895e7a1d75489903dcfa2efe3e2e738c6fe454 Mon Sep 17 00:00:00 2001 From: Administrator Date: Fri, 15 Sep 2023 08:28:57 +0800 Subject: [PATCH] add imports --- .../java/ai/djl/examples/inference/Yolov8Detection.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/src/main/java/ai/djl/examples/inference/Yolov8Detection.java b/examples/src/main/java/ai/djl/examples/inference/Yolov8Detection.java index 7dcc88e9a9a..e8ddb6e8a00 100644 --- a/examples/src/main/java/ai/djl/examples/inference/Yolov8Detection.java +++ b/examples/src/main/java/ai/djl/examples/inference/Yolov8Detection.java @@ -16,11 +16,15 @@ import ai.djl.inference.Predictor; import ai.djl.modality.cv.Image; import ai.djl.modality.cv.ImageFactory; +import ai.djl.modality.cv.output.BoundingBox; import ai.djl.modality.cv.output.DetectedObjects; -import ai.djl.modality.cv.translator.YoloV5TranslatorFactory; +import ai.djl.modality.cv.output.DetectedObjects.DetectedObject; +import ai.djl.modality.cv.output.Rectangle; +import ai.djl.modality.cv.translator.YoloV8TranslatorFactory; import ai.djl.repository.zoo.Criteria; import ai.djl.repository.zoo.ZooModel; import ai.djl.training.util.ProgressBar; +import ai.djl.translate.Translator; import ai.djl.translate.TranslateException; import org.slf4j.Logger; @@ -30,6 +34,7 @@ import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; +import java.util.List; /** * An example of inference using an yolov8 model.