change default setting

google · Apr 18, 2022 · eda6c82 · eda6c82
1 parent c5231d2
commit eda6c82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/efficientdet/tf2/postprocess.py b/efficientdet/tf2/postprocess.py
@@ -183,7 +183,7 @@ def nms(params, boxes: T, scores: T, classes: T,
     score_thresh = nms_configs['score_thresh'] or float('-inf')
   elif method == 'gaussian':
     sigma = nms_configs['sigma'] or 0.5
-    iou_thresh = 1.0
+    iou_thresh = 0.5
     score_thresh = nms_configs['score_thresh'] or 0.001
   else:
     raise ValueError('Inference has invalid nms method {}'.format(method))