yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
4cc109ec9f
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  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')