基于YOLOv5改进算法的印花图案疵点检测

被引:18
作者
颜学坤
楚建安
机构
[1] 西安工程大学电子信息学院
关键词
印花疵点; YOLOv5; 注意力机制;
D O I
10.19651/j.cnki.emt.2108370
中图分类号
TS194.4 [印花方法]; TP391.41 [];
学科分类号
080203 ;
摘要
针对圆网印花图案疵点检测问题,采用了一种基于YOLOv5改进算法模型来检测印花图案的疵点。根据实际的情况对YOLOv5模型网络结构进行了更改,首先,对YOLOv5网络的骨干部分进行优化改进,引入了注意力机制模块,对输入图片的通道注意力和空间注意分别提取特征。其次,针对印花疵点目标较小的情况对网络的检测层结构进行了修改。实验结果显示,改进的YOLOv5检测算法精确率提升了14.4%,检测速度提升了7.6 fps,达到了43.1 fps满足实时检测要求。
引用
收藏
页码:59 / 65
页数:7
相关论文
共 5 条