173、MixUp 数据增强在 YOLOv11 中的实现:双图混合的 Beta 分布 α 参数消融

发布时间:2026/7/10 20:05:26
173、MixUp 数据增强在 YOLOv11 中的实现:双图混合的 Beta 分布 α 参数消融 173、MixUp 数据增强在 YOLOv11 中的实现:双图混合的 Beta 分布 α 参数消融上周调一个工地安全帽检测项目,发现模型对小目标头盔的召回率死活上不去。试了Mosaic、CutMix,效果都一般。后来翻到一篇老论文《mixup: Beyond Empirical Risk Minimization》,突然意识到——YOLOv11官方居然没内置MixUp。自己动手补上,顺便把α参数从0.1到2.0跑了个遍,发现这玩意儿对YOLO系模型的收敛行为影响比想象中大得多。为什么YOLOv11需要MixUpYOLOv11的Mosaic增强虽然强,但本质是空间拼接,四个图拼一起,边界处容易产生不自然的语义断裂。MixUp不一样,它是像素级线性插值,两张图按比例混合,标签也按同样比例混合。这种软标签训练方式对YOLO的密集预测头特别友好——不会像Mosaic那样让某些anchor突然面对完全不存在的目标。我踩过的坑:直接套用ImageNet分类的MixUp代码到YOLOv11,训练到一半loss直接炸了。原因在于YOLO的loss计算涉及正负样本匹配,混合标签的iou计算需要特殊处理。代码实现:在YOLOv11的DataLoader中注入MixUp第一步:修改数据集加载逻辑在ultralytics/data/dataset.py中,找到YOLOData