Browse Source

debug YOLOX-style Transform with Rotation

yjh0410 2 years ago
parent
commit
e381560457
2 changed files with 2 additions and 2 deletions
  1. 1 1
      eval.py
  2. 1 1
      train.py

+ 1 - 1
eval.py

@@ -32,7 +32,7 @@ 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('-ct', '--conf_thresh', default=0.001, type=float,
+    parser.add_argument('-ct', '--conf_thresh', default=0.005, type=float,
                         help='confidence threshold')
     parser.add_argument('-nt', '--nms_thresh', default=0.6, type=float,
                         help='NMS threshold')

+ 1 - 1
train.py

@@ -61,7 +61,7 @@ def parse_args():
     # Model
     parser.add_argument('-m', '--model', default='yolov1', type=str,
                         help='build yolo')
-    parser.add_argument('-ct', '--conf_thresh', default=0.001, type=float,
+    parser.add_argument('-ct', '--conf_thresh', default=0.005, type=float,
                         help='confidence threshold')
     parser.add_argument('-nt', '--nms_thresh', default=0.6, type=float,
                         help='NMS threshold')