NMS_IOU

counter_live_rknn.py

@@ -39,6 +39,7 @@ LINE_Y2_FRAC = float(os.getenv("LINE_Y2_FRAC", "0.66"))
 IMGSZ = int(os.getenv("IMGSZ", "320"))
 HALF = os.getenv("HALF", "false").lower() == "true"
 CONF = float(os.getenv("CONF", "0.3"))
+NMS_IOU = float(os.getenv("NMS_IOU", "0.45"))
 DEVICE = int(os.getenv("DEVICE", "0"))
 
 # RKNN NPU core mask
@@ -1055,6 +1056,7 @@ def run():
         core_mask=CORE_MASK,
         imgsz=IMGSZ,
         conf=CONF,
+        iou=NMS_IOU,
         num_classes=NUM_CLASSES,
         score_sigmoid=SCORE_SIGMOID,
     )