Added v8 parameter to edgetpumodel.py and nms.py

sebastianfis · Mar 24, 2024 · 18f6fa1 · 18f6fa1
1 parent 54a1c54
commit 18f6fa1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nms.py b/nms.py
@@ -93,9 +93,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
             continue
 
         # Compute conf
-        if v8:
-            x[:, 4:] *= x[:, 4]  
-        else:
+        if not v8:
             x[:, 5:] *= x[:, 4:5]  # conf = obj_conf * cls_conf
 
         # Box (center x, center y, width, height) to (x1, y1, x2, y2)