标题:基于YOLO-EPWC算法的新能源汽车轮胎缺陷检测 作者:杨海燕 崔文君 贾岩龙 作者单位:天津职业技术师范大学电子工程学院,天津 300222 关键字:YOLOv8;新能源汽车轮胎缺陷;EfficientRep;极化自注意力;Wise-IoU;CELU 摘要:针对新能源汽车轮胎X光图像缺陷问题,构建一种基于YOLOv8的缺陷检测算法。首先,使用EfficientRep网络替换YOLOv8原始主干网络,有效平衡执行精度和速度;其次,将极化自注意力机制(PSA)引入YOLOv8头部保持图像的高分辨输出,提升定位准确性;最后,将CIoU损失函数和SiLU激活函数分别替换为Wise-IoU和CELU,提高收敛能力和检测精度,使训练过程更加稳定。实验结果表明,相较于原YOLOv8n模型,YOLO-EPWC在图像检测的平均精度mAP0.5、mAP0.5:0.95、召回率(R)和精度(P)上,分别提升了4.0%、3.4%、4.3%、1.0%。该算法可以较好地改善传统新能源汽车轮胎缺陷检测所面临的问题。 |