Преглед на файлове

modify RT-PlainDETR's Trainer

yjh0410 преди 1 година
родител
ревизия
5495a4c2eb
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      train.py

+ 1 - 1
train.py

@@ -101,7 +101,7 @@ def parse_args():
                         help='Multi scale')
     parser.add_argument('--ema', action='store_true', default=False,
                         help='Model EMA')
-    parser.add_argument('--min_box_size', default=1.0, type=float,
+    parser.add_argument('--min_box_size', default=8.0, type=float,
                         help='min size of target bounding box.')
     parser.add_argument('--mosaic', default=None, type=float,
                         help='mosaic augmentation.')