首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进YOLOv5算法的纸袋缺陷检测
作者姓名:杨萌  张爱军  潘文松
作者单位:1. 南京理工大学机械工程学院;2. 杭州康代思锐生物科技有限公司
基金项目:江苏省研究生科研与实践创新计划项目(SJCX22_0103);
摘    要:为了提高纸袋生产企业在制造过程中对纸袋手把或底部缺陷的检测精度,提出了一种基于改进YOLOv5算法的纸袋缺陷检测方法。改进算法为了提高网络定位能力,增强网络的特征学习表达能力,引入了坐标注意力机制,接着引入EIoU损失函数对原始损失函数进行改进,以此来改善原始网络损失函数纵横比的合理性,提升回归精度,最后引入一种具有类似跨阶段局部结构的简化空间金字塔池化结构,减少冗余信息处理,提升网络检测性能。实验结果表明,改进算法的平均精度平均值mAP@.5为87.3%,mAP@.5∶.95为56.8%,与YOLOv5算法相比mAP@.5提升了1.6%,mAP@.5∶.95提升了0.9%,在纸袋缺陷检测上有更优越的表现。

关 键 词:纸袋缺陷  注意力  损失函数  空间金字塔池化  改进算法
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号