【目标检测论文解读复现NO.25】基于改进Yolov5的地铁隧道附属设施与衬砌表观病害检测方法

阿里云国内75折回扣微信号：monov8

阿里云国际，腾讯云国际，低至75折。AWS 93折免费开户实名账号代冲值优惠多多微信号：monov8 飞机：@monov6

前言

此前出了目标改进算法专栏但是对于应用于什么场景需要什么改进方法对应与自己的应用场景有效果并且多少改进点能发什么水平的文章为解决大家的困惑此系列文章旨在给大家解读最新目标检测算法论文帮助大家解答疑惑。解读的系列文章本人已进行创新点代码复现有需要的朋友可关注私信我。

一、摘要

针对目前应用到织物疵点检测的网络模型中存在的两阶段算法检测速度慢、一阶段算法检测精度低的问题提出了一种改进YOLOv5的织物疵点检测算法。首先针对织物疵点大小不一对K-Mean算法的聚类距离标准进行修改重新计算先验框大小其次对网络Neck层标准卷积SC进行改进将深度分离卷积DSC与标准卷积结合减少网络层参数量同时保持网络的特征提取能力在特征融合阶段引入坐标注意力机制CA使网络能够捕捉各通道之间联系的同时保留目标的精确定位信息加强网络的特征提取和定位能力最后使用加权双向特征金字塔网络BiFPN中的方法对特征金字塔模块进行修改实现简单快速的多尺度特征融合。在数据集上进行训练结果表明改进的YOLOv5模型的m AP值可达到97.4%相比于原网络精度提高了2.8个百分点满足了织物疵点检测的要求。

二、网络模型及核心创新点

个人解读这篇1月份发表的文章的创新点虽然不是很新其中的GSConv比较少见其他几个改进方法都比较常见了但是其检测应用场景比较少见可能也是能被核心期刊录用的重要原因之一另外文章中的图表很多实验工作量也较多。相关创新点改进方法及代码也已经在之前的博文中进行介绍可进行手动链接查看。需要将几种合为一起的方法可私信我获取。

K-Mean算法的聚类距离标准进行修改 [YOLOv7/YOLOv5系列算法改进NO.9]锚框K-Means算法改进K-Means++_yolov5聚类优化
深度分离卷积DSC与标准卷积结合GSConv【YOLOv7/YOLOv5系列算法改进NO.48】构建新的轻量网络—Slim-neck by GSConv2022CVPR
坐标注意力机制[YOLOv7/YOLOv5系列算法改进NO.3]添加CoordAtt注意力机制
加权双向特征金字塔网络 [YOLOv7/YOLOv5系列算法改进NO.5]改进特征融合网络PANET为BIFPN更新添加小目标检测层yaml

三、应用数据集

本文研究的织物疵点图像来自于阿里天池纺织品缺陷检测数据集和实验室内拍摄的疵点图像图像尺寸不一后经过处理后将数据集中的所有图像调整为640像素×640像素然后对图像进行随机旋转变化、亮度调节、色彩变换及自适应对比增强等预处理。

四、实验效果消融实验

最终网络改进效果如表4的改进6所示实验结果表明网络中加入各种改进后各项指标都有所提升最终改进的网络精度提升2.4%召回率提高2.5%m AP提升了2.8%达到了97.4%的效果。图11为改进6与原网络的损失函数收敛曲线和精度收敛曲线的对比情况可以看出相比于原网络而言经过200轮的训练改进后的网络loss函数收敛速度更快并且最终loss值更低。说明改进后的网络性能更优秀。其中损失函数使用CIOU进行计算CIOU考虑了重叠面积、中心距离、纵横比在DIOU的基础上增加了检测框尺度的loss增加了长和宽的loss使得预测框更加符合真实框。

五、实验结论

本文针对织物疵点复杂多样传统检测方法检测速度慢精度低等问题提出一种改进YOLOv5s的织物疵点检测算法。该方法基于YOLOv5s-6.1网络构架其网络相比主流网络更为轻量化检测速度较快本文将CA注意力机制与网络的部分Neck层结合同时使用BiFPN的方法对网络的特征增强部分重新设计然后再使用GSConv对网络的特征提取部分进一步去除冗余特征保证了网络的
精度与速度。实验表明本文提出的疵点检测方法能够准确快速的检测出常见的织物疵点图像在NVIDIA RTX 3090上测试平均单张疵点图像检测速度可达12.0msmAP达到了97.4%综合性能明显高于常见目标检测网络。