Ver código fonte

modify RT-PlainDETR's Trainer

yjh0410 1 ano atrás
pai
commit
5495a4c2eb
1 arquivos alterados com 1 adições e 1 exclusões
  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.')