update

Author: yjh0410

config/__init__.py

@@ -91,8 +91,6 @@ def build_trans_config(trans_config='ssd'):
 from .model_config.yolov7_config import yolov7_cfg
 from .model_config.yolov8_config import yolov8_cfg
 from .model_config.yolox_config  import yolox_cfg
-## Real-time DETR series
-from .model_config.rtdetr_config import rtdetr_cfg
 
 def build_model_config(args):
     print('==============================')
@@ -124,9 +122,6 @@ def build_model_config(args):
     # YOLOX
     elif args.model in ['yolox_n', 'yolox_s', 'yolox_m', 'yolox_l', 'yolox_x']:
         cfg = yolox_cfg[args.model]
-    # RT-DETR
-    elif args.model in ['rtdetr_r18', 'rtdetr_r34', 'rtdetr_r50', 'rtdetr_r101']:
-        cfg = rtdetr_cfg[args.model]
 
     return cfg
 

config/model_config/rtdetr_config.py

@@ -11,8 +11,6 @@
 from .yolov7.build import build_yolov7
 from .yolov8.build import build_yolov8
 from .yolox.build import build_yolox
-# Real-time DETR series
-from .rtdetr.build import build_rtdetr
 
 
 # build object detector
@@ -62,10 +60,6 @@ def build_model(args,
     elif args.model in ['yolox_n_adamw', 'yolox_s_adamw', 'yolox_m_adamw', 'yolox_l_adamw', 'yolox_x_adamw']:
         model, criterion = build_yolox(
             args, model_cfg, device, num_classes, trainable, deploy)
-    # RT-DETR
-    elif args.model in ['rtdetr_r18', 'rtdetr_r34', 'rtdetr_r50', 'rtdetr_r101']:
-        model, criterion = build_rtdetr(
-            args, model_cfg, num_classes, trainable, deploy)
 
     if trainable:
         # Load pretrained weight