lower iou threshold

nobleo · Sep 25, 2024 · 12146d6 · 12146d6
1 parent d6186dc
commit 12146d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rtdetrv2_pytorch/src/nn/postprocessor/nms_postprocessor.py b/rtdetrv2_pytorch/src/nn/postprocessor/nms_postprocessor.py
@@ -18,7 +18,7 @@
 @register()
 class DetNMSPostProcessor(torch.nn.Module):
     def __init__(self, \
-                iou_threshold=0.7, 
+                iou_threshold=0.01, 
                 score_threshold=0.1, 
                 keep_topk=300, 
                 box_fmt='cxcywh',