Browse Source

modify conf_thresh

冬落 2 năm trước cách đây
mục cha
commit
a27921eb31
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      demo.py

+ 1 - 1
demo.py

@@ -54,7 +54,7 @@ def parse_args():
                         help='number of classes.')
     parser.add_argument('--weight', default=None,
                         type=str, help='Trained state_dict file path to open')
-    parser.add_argument('-ct', '--conf_thresh', default=0.1, type=float,
+    parser.add_argument('-ct', '--conf_thresh', default=0.3, type=float,
                         help='confidence threshold')
     parser.add_argument('-nt', '--nms_thresh', default=0.5, type=float,
                         help='NMS threshold')