diff --git a/test.py b/test.py index d7cc4294..1be99214 100644 --- a/test.py +++ b/test.py @@ -14,7 +14,7 @@ def test(cfg, batch_size=16, img_size=416, conf_thres=0.001, - iou_thres=0.5, # for nms + iou_thres=0.6, # for nms save_json=False, single_cls=False, model=None,