retrain YOLOX on VOC

config/transform_config.py

@@ -58,64 +58,6 @@ yolov5_nano_trans_config = {
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp
 }
 
-# ----------------------- YOLOX-Style -----------------------
-yolox_strong_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': 'yolox_mixup',
-    'mixup_scale': [0.5, 1.5]
-}
-
-yolox_weak_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.15,
-    'mosaic_type': 'yolov5_mosaic',
-    'mixup_type': 'yolox_mixup',
-    'mixup_scale': [0.5, 1.5]
-}
-
-yolox_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': 0.5,
-    'mixup_prob': 0.0,
-    'mosaic_type': 'yolov5_mosaic',
-    'mixup_type': 'yolox_mixup',
-    'mixup_scale': [0.5, 1.5]
-}
-
 # ----------------------- SSD-Style -----------------------
 ssd_trans_config = {
     'aug_type': 'ssd',

config/yolox_config.py

@@ -2,7 +2,7 @@
 
 yolox_cfg = {
     # input
-    'trans_type': 'yolox_strong',
+    'trans_type': 'yolov5_strong',
     'multi_scale': [0.5, 1.0],
     # model
     'backbone': 'cspdarknet',

train.sh

@@ -3,7 +3,7 @@ python train.py \
         --cuda \
         -d voc \
         --root /mnt/share/ssd2/dataset/ \
-        -m yolov7_large \
+        -m yolox \
         -bs 16 \
         -size 640 \
         --wp_epoch 1 \