update

config/__init__.py

@@ -31,7 +31,14 @@ from .data_config.transform_config import (
     yolox_large_trans_config,
     yolox_huge_trans_config,
     # SSD-Style
-    ssd_trans_config
+    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'):
@@ -69,7 +76,21 @@ def build_trans_config(trans_config='ssd'):
         cfg = yolox_large_trans_config
     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
 
 

config/data_config/transform_config.py

@@ -240,3 +240,112 @@ ssd_trans_config = {
     'mixup_type': 'yolov5_mixup',
     'mixup_scale': [0.5, 1.5]
 }
+
+# ----------------------- RTMDet-Style Transform -----------------------
+rtmdet_huge_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.2,
+    'scale': 0.9,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 1.0,
+    'mixup_prob': 1.0,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}
+
+rtmdet_large_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.2,
+    'scale': 0.9,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 1.0,
+    'mixup_prob': 1.0,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}
+
+rtmdet_medium_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.2,
+    'scale': 0.9,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 1.0,
+    'mixup_prob': 1.0,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}
+
+rtmdet_small_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.2,
+    'scale': 0.9,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 1.0,
+    'mixup_prob': 1.0,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}
+
+rtmdet_nano_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.1,
+    'scale': 0.5,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 1.0,
+    'mixup_prob': 0.5,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}
+
+rtmdet_pico_trans_config = {
+    'aug_type': 'yolov5',
+    # Basic Augment
+    'degrees': 0.0,
+    'translate': 0.1,
+    'scale': 0.5,
+    'shear': 0.0,
+    'perspective': 0.0,
+    'hsv_h': 0.015,
+    'hsv_s': 0.7,
+    'hsv_v': 0.4,
+    # Mosaic & Mixup
+    'mosaic_prob': 0.5,
+    'mixup_prob': 0.0,
+    'mosaic_type': 'yolov5_mosaic',
+    'mixup_type': 'yolov5_mixup',
+}