train YOLOvx-M on COCO

config/__init__.py

@@ -32,13 +32,6 @@ from .data_config.transform_config import (
     yolox_huge_trans_config,
     # SSD-Style
     ssd_trans_config,
-    # RTMDet-Style
-    rtmdet_pico_trans_config,
-    rtmdet_nano_trans_config,
-    rtmdet_small_trans_config,
-    rtmdet_medium_trans_config,
-    rtmdet_large_trans_config,
-    rtmdet_huge_trans_config
 )
 
 def build_trans_config(trans_config='ssd'):
@@ -77,20 +70,6 @@ def build_trans_config(trans_config='ssd'):
     elif trans_config == 'yolox_huge':
         cfg = yolox_huge_trans_config
 
-    # RTMDet-style transform
-    elif trans_config == 'rtmdet_pico':
-        cfg = rtmdet_pico_trans_config
-    elif trans_config == 'rtmdet_nano':
-        cfg = rtmdet_nano_trans_config
-    elif trans_config == 'rtmdet_small':
-        cfg = rtmdet_small_trans_config
-    elif trans_config == 'rtmdet_medium':
-        cfg = rtmdet_medium_trans_config
-    elif trans_config == 'rtmdet_large':
-        cfg = rtmdet_large_trans_config
-    elif trans_config == 'rtmdet_huge':
-        cfg = rtmdet_huge_trans_config
-
     return cfg
 
 

test.py

@@ -35,6 +35,8 @@ def parse_args():
                         help='Final confidence threshold')
     parser.add_argument('-ws', '--window_scale', default=1.0, type=float,
                         help='resize window of cv2 for visualization.')
+    parser.add_argument('--resave', action='store_true', default=False, 
+                        help='resave checkpoints without optimizer state dict.')
 
     # model
     parser.add_argument('-m', '--model', default='yolov1', type=str,