当前位置: X-MOL 学术Sci. Rep. › 论文详情
Our official English website, www.x-mol.net, welcomes your feedback! (Note: you will need to create a separate account there.)

基于改进YOLOv7的轻量化带钢缺陷检测算法

Scientific Reports ( IF 3.8 ) Pub Date : 2024-06-10 , DOI: 10.1038/s41598-024-64080-x
Jianbo Lu 1 , MiaoMiao Yu 2 , Junyu Liu 1
Affiliation  


准确识别带钢表面缺陷对于保证钢产品质量至关重要。为了解决当前带钢表面缺陷检测算法中模型尺寸大和计算复杂性带来的挑战,本文引入了SS-YOLO(YOLOv7 for Steel Strip),这是一种增强的轻量级YOLOv7模型。该方法用轻量级的MobileNetv3网络取代了主干网络中的CBS模块,减少了模型大小并加快了推理时间。 D-SimSPPF模块集成了深度可分离卷积和无参数注意力机制,专门用于替换YOLOv7网络内原有的SPPCSPC模块,扩大感受野并减少网络参数数量。在颈部网络和预测输出部分均引入了无参数注意力机制SimAM,增强了模型提取带材表面缺陷本质特征的能力,提高了检测精度。 NEU-DET数据集上的实验结果表明,SS-YOLO达到了97%的mAP50准确率,比YOLOv7提​​高了4.5%。此外,FLOPs(G) 减少了 79.3%,参数减少了 20.7%。因此,SS-YOLO 在保持轻量级外形的同时,展示了检测精度和速度之间的有效平衡。





"点击查看英文标题和摘要"

更新日期:2024-06-10
down
wechat
bug