From 18f6fa178bac7163dc70795f186f7adec0495c03 Mon Sep 17 00:00:00 2001
From: Sebastian <sebastian_fis@yahoo.de>
Date: Sun, 24 Mar 2024 01:23:58 +0100
Subject: [PATCH] Added v8 parameter to edgetpumodel.py and nms.py

---
 nms.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/nms.py b/nms.py
index 541a642..b20e2f0 100644
--- 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)