Bladeren bron

debug YOLOv7

yjh0410 2 jaren geleden
bovenliggende
commit
b3c4b445e4
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      eval.py

+ 1 - 1
eval.py

@@ -31,7 +31,7 @@ def parse_args():
                         help='build yolo')
                         help='build yolo')
     parser.add_argument('--weight', default=None,
     parser.add_argument('--weight', default=None,
                         type=str, help='Trained state_dict file path to open')
                         type=str, help='Trained state_dict file path to open')
-    parser.add_argument('--conf_thresh', default=0.005, type=float,
+    parser.add_argument('--conf_thresh', default=0.001, type=float,
                         help='NMS threshold')
                         help='NMS threshold')
     parser.add_argument('--nms_thresh', default=0.6, type=float,
     parser.add_argument('--nms_thresh', default=0.6, type=float,
                         help='NMS threshold')
                         help='NMS threshold')