Ver Fonte

modify YOLOvx-Small-Aug

yjh0410 há 2 anos atrás
pai
commit
d41dc6a7be
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      config/data_config/transform_config.py

+ 1 - 1
config/data_config/transform_config.py

@@ -312,7 +312,7 @@ yolovx_small_trans_config = {
     'hsv_v': 0.4,
     # Mosaic & Mixup
     'mosaic_prob': 1.0,
-    'mixup_prob': 1.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