fix bug

deepjavalibrary · Feb 10, 2024 · 49e071d · 49e071d
1 parent be4d1d3
commit 49e071d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/java/ai/djl/modality/cv/translator/YoloV8Translator.java b/api/src/main/java/ai/djl/modality/cv/translator/YoloV8Translator.java
@@ -75,7 +75,7 @@ protected DetectedObjects processFromBoxOutput(NDList list) {
                 float classProb = buf[index + c];
                 if (classProb > maxClassProb) {
                     maxClassProb = classProb;
-                    maxIndex = c;
+                    maxIndex = c - 4;
                 }
             }