为满足多数工业场景下钢板表面缺陷检测的需求,针对钢板表面缺陷检测准确率低及小目标缺陷检测率低等问题,文中提出了一种基于改进YOLOv5(You Only Look Once version 5)的钢板表面缺陷检测算法。在YOLOv5的基础上将CBAM(Convolution Bl...为满足多数工业场景下钢板表面缺陷检测的需求,针对钢板表面缺陷检测准确率低及小目标缺陷检测率低等问题,文中提出了一种基于改进YOLOv5(You Only Look Once version 5)的钢板表面缺陷检测算法。在YOLOv5的基础上将CBAM(Convolution Block Attention Module)注意力模块嵌入到主干网络中,提高网络检测精度。加入上下文增强模块,提高了算法对小目标的检测性能。使用NWD(Normalized Wasserstein Distance)度量标准代替原YOLOv5中的IoU(Intersection over Union)度量,提高了网络对裂纹缺陷的识别精确度。实验结果表明,钢板表面缺陷检测算法对裂纹、夹杂、斑块、麻点、压入氧化铁皮、划痕6类缺陷的平均检测精度达到了88.9%,每秒帧数达到110.4 frame·s-1,其中小目标裂纹准确率达到75%。展开更多
精准识别和定位飞机目标是航空安全和信息化战争胜利的关键,针对传统飞机目标识别抗干扰性差,对遮挡、光照、尺度敏感难应对复杂场景需求的问题,提出了一种基于改进YOLOv5的飞机目标检测算法。通过IOU-NWD Similarity Metric for Boundi...精准识别和定位飞机目标是航空安全和信息化战争胜利的关键,针对传统飞机目标识别抗干扰性差,对遮挡、光照、尺度敏感难应对复杂场景需求的问题,提出了一种基于改进YOLOv5的飞机目标检测算法。通过IOU-NWD Similarity Metric for Bounding Boxes策略解决了IOU机制对飞机小目标的标签分配歧义问题;使用GFPN based on NLnet模块完成了“跨层”与“跨尺度”的自适应融合,更加丰富和具有代表性的特征信息;使用soft-NMS方法解决了在目标密集区域中飞机小目标存在的漏检问题。在飞机数据集上进行实验,结果表明:与原始YOLOv5相比,改进后的模型在Precision、Recall、mAP0.5、mAP0.5:0.95分别提高了1.9%、10.4%、3.6%和5.8%。该算法通过针对性的网络调整和模块迁移来提高模型对小型和遮挡的飞机目标的检测效果,并通过实验验证了该算法的优越性,实验结果表明,AIR-YOLO在检测精度和鲁棒性方面优于YOLOv5,解决了原始YOLOv5算法的小飞机目标误检的问题。展开更多
文摘为满足多数工业场景下钢板表面缺陷检测的需求,针对钢板表面缺陷检测准确率低及小目标缺陷检测率低等问题,文中提出了一种基于改进YOLOv5(You Only Look Once version 5)的钢板表面缺陷检测算法。在YOLOv5的基础上将CBAM(Convolution Block Attention Module)注意力模块嵌入到主干网络中,提高网络检测精度。加入上下文增强模块,提高了算法对小目标的检测性能。使用NWD(Normalized Wasserstein Distance)度量标准代替原YOLOv5中的IoU(Intersection over Union)度量,提高了网络对裂纹缺陷的识别精确度。实验结果表明,钢板表面缺陷检测算法对裂纹、夹杂、斑块、麻点、压入氧化铁皮、划痕6类缺陷的平均检测精度达到了88.9%,每秒帧数达到110.4 frame·s-1,其中小目标裂纹准确率达到75%。
文摘精准识别和定位飞机目标是航空安全和信息化战争胜利的关键,针对传统飞机目标识别抗干扰性差,对遮挡、光照、尺度敏感难应对复杂场景需求的问题,提出了一种基于改进YOLOv5的飞机目标检测算法。通过IOU-NWD Similarity Metric for Bounding Boxes策略解决了IOU机制对飞机小目标的标签分配歧义问题;使用GFPN based on NLnet模块完成了“跨层”与“跨尺度”的自适应融合,更加丰富和具有代表性的特征信息;使用soft-NMS方法解决了在目标密集区域中飞机小目标存在的漏检问题。在飞机数据集上进行实验,结果表明:与原始YOLOv5相比,改进后的模型在Precision、Recall、mAP0.5、mAP0.5:0.95分别提高了1.9%、10.4%、3.6%和5.8%。该算法通过针对性的网络调整和模块迁移来提高模型对小型和遮挡的飞机目标的检测效果,并通过实验验证了该算法的优越性,实验结果表明,AIR-YOLO在检测精度和鲁棒性方面优于YOLOv5,解决了原始YOLOv5算法的小飞机目标误检的问题。
