modify YOLOX-Nano Transform

config/data_config/transform_config.py

@@ -207,7 +207,7 @@ yolox_nano_trans_config = {
     'hsv_v': 0.4,
     # Mosaic & Mixup
     'mosaic_prob': 1.0,
-    'mixup_prob': 0.0,
+    'mixup_prob': 0.5,
     'mosaic_type': 'yolov5_mosaic',
     'mixup_type': 'yolox_mixup',
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp

train_multi_gpus.sh

@@ -4,7 +4,7 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     -dist \
                                                     -d coco \
                                                     --root /data/datasets/ \
-                                                    -m rtcdet_t \
+                                                    -m rtcdet_n \
                                                     -bs 128 \
                                                     -size 640 \
                                                     --wp_epoch 3 \