基于CSE-YOLO的改进铝型材表面缺陷检测模型OA
Improved Surface Defect Detection Model for Aluminum Profile Based on CSE-YOLO
针对提升铝型材表面缺陷检测精度时会导致模型的复杂度、检测速度与精度难以平衡的问题,提出一种基于改进YOLOv5的检测模型CSE-YOLO.首先,结合CCFF与BiFPN建立新的网络结构CCFBiFPN,以加强多尺度特征融合,降低模型的复杂度;其次,在特征融合模块引入混合GSConv的SDI模块,提高模型的特征提取能力;最后,在模型主干网络中增加ECA注意力机制,提高模型的精确度与总体性能.实验结果表明,改进模型相较于传统YO-LOv5s模型在mAP@0.5和mAP@0.5:0.9指标上分别提升了5%与1.3%.同时,模型的参数量和浮点运算次数分别降低了4.3%与4.4%,且每秒检测帧数从140提升到了143.该方法有效平衡了模型的性能与复杂度,具有较高的工程应用价值.
To address the challenge of balancing model complexity,detection speed,and accuracy in aluminum profile surface defect detec-tion,an enhanced YOLOv5-based model named CSE-YOLO was developed.First,a novel CCFBiFPN architecture was constructed by inte-grating CCFF with BiFPN,which strengthened multi-scale feature fusion while reducing model complexity.Second,an SDI module incorporat-ing hybrid GSConv was implemented in the feature fusion stage to improve feature extraction capability.Finally,an ECA attention mechanism was embedded in the backbone network to enhance detection precision and overall performance.Experimental results demonstrated that com-pared with the baseline YOLOv5s model,the proposed approach achieved 5%and 1.3%improvements in mAP@0.5 and mAP@0.5:0.9 met-rics,respectively.Concurrently,parameter count and FLOPs were reduced by 4.3%and 4.4%,with the frame rate increasing from 140 to 143 FPS.The optimized architecture effectively balanced performance and complexity,demonstrating substantial practical value for industrial qual-ity inspection applications.
刘子龙;庞明;窦建明
兰州交通大学 机电工程学院,甘肃 兰州 730070兰州交通大学 机电工程学院,甘肃 兰州 730070兰州交通大学 机电工程学院,甘肃 兰州 730070
信息技术与安全科学
铝型材表面缺陷检测YOLOv5注意力机制特征融合
aluminum profilesurface defect detectionYOLOv5attention mechanismfeature fusion
《软件导刊》 2026 (4)
98-103,6
评论