yjh0410 2 vuotta sitten
vanhempi
sitoutus
4cc109ec9f
1 muutettua tiedostoa jossa 3 lisäystä ja 3 poistoa
  1. 3 3
      eval.py

+ 3 - 3
eval.py

@@ -31,9 +31,9 @@ def parse_args():
                         help='build yolo')
     parser.add_argument('--weight', default=None,
                         type=str, help='Trained state_dict file path to open')
-    parser.add_argument('--conf_thresh', default=0.001, type=float,
-                        help='NMS threshold')
-    parser.add_argument('--nms_thresh', default=0.6, type=float,
+    parser.add_argument('-ct', '--conf_thresh', default=0.001, type=float,
+                        help='confidence threshold')
+    parser.add_argument('-nt', '--nms_thresh', default=0.6, type=float,
                         help='NMS threshold')
     parser.add_argument('--topk', default=1000, type=int,
                         help='topk candidates for testing')