关键词:
疵点检测
YOLOv8算法
GhostNet
坐标注意力
BiFPN
摘要:
针对织物疵点检测中疵点形态各异,检测结果容易存在漏检或误检等问题,提出一种改进YOLOv8的织物疵点检测算法。首先将主干网络替换为GhostNet,该方法在减少对计算资源要求的同时,在保证网络结构质量的前提下,减轻了网络结构的重量。另外,该模型还利用了一系列有效的线性运算,减少了模型中的参数,获得更多的特征图谱,减少了算法的计算量,大大提高了模型的性能。其次引入CA注意力机制(coordinate attention),有效增强了网络的特征提取能力。最后引入BiFPN金字塔替换head层中的concat连接,将语义信息传递到不同的特征尺度上,从而增强特征融合。实验结果表明改进的算法在检测8类织物疵点时,mAP@0.5达到91.9%,mAP@0.5:0.95达到48.8%,相比原始的YOLOv8n算法分别提高了2.2%和1.1%,具备更高的检测精度,具有更少的漏检或误检情况。