来源:期刊VIP网 时间:
作者:李西兴;刘涛;周宏娣;吴锐;陈佳豪;
单位:湖北工业大学机械工程学院;
摘要:针对瓦楞纸板表面缺陷检测速度慢和识别准确率低等问题,提出基于改进YOLOv5s的轻量化瓦楞纸板表面缺陷检测算法YOLOv5s-GCS。将YOLOv5s骨干网络中原有的Conv模块替换为GhostConv模块,用C2f模块替换C3模块,并集成置换注意力机制(SA)模块;在YOLOv5s颈部网络末端引入SA模块;通过构建瓦楞纸板表面缺陷数据集进行试验验证。试验结果显示,YOLOv5s-GCS算法平均精度均值达到95.0%、召回率达到89.2%、精确率达到92.5%,较原始YOLOv5s分别提高2.3%,1.3%,2.8%;检测速度达到19.9帧/s,较原始YOLOv5s提高5.7帧/s。YOLOv5s-GCS算法更有利于迁移部署与实际应用。研究为表面缺陷领域的实时检测提供参考。
关键词:YOLOv5s;;表面缺陷检测;;置换注意力机制;;GhostConv模块;;C2f模块
基金资助:国家自然科学基金项目(51805152);; 湖北省自然科学基金项目(2024AFB816);; 湖北省重点研发计划项目(2023BEB043)