基于YOLOv5n-CND的矿用输送带异物检测
孙奥然1
赵培培1
杨迪1
张君逸1
于洪健2
1.中国矿业大学信息与控制工程学院,江苏徐州 2211162.华北理工大学人工智能学院,河北唐山 063009
摘要：针对异物图像背景复杂、特征提取能力弱、粘连小目标的检测精度低、检测框定位及尺度失真等问题,提出了一种基于YOLOv5n-CND的矿用输送带目标检测算法.首先,采用C2f对特征金字塔进行优化,使用更少参数解决在井下异物图像采集背景复杂且存在复杂目标干扰对小目标检测不敏感的问题;然后,采用归一化高斯瓦萨斯坦距离(NWD)回归损失函数替代CIoU,改善多尺度异物检测效果不佳的问题,实现粘连小目标的精准检测;最后,添加目标检测头(Dy Head),将尺度、空间和任务3种注意力机制结合,提高对异物轮廓的特征提取能力,增强对多尺度目标的适应能力.实验结果表明:YOLOv5n-CND的mAP@0.5、mAP@0.5∶0.95、参数量及检测速度分别为87.9％,55.9％,4.49×106个,85.5帧/s,满足煤矿井下异物检测需求;YOLOv5n-CND的mAP@0.5和mAP@0.5∶0.95 较 YOLOv5n 分别提高了 2.6％和 3.4％,较 YOLOv5s-CBAM 分别提高了 1.7％和 3.8％;模型参数量在YOLOv5n的基础上略有提升,但较其他模型参数量均有所降低.选取异物与背景相近的细长检测物、光照比较低的锚杆检测物、大量煤矸石混杂的检测物、含有多个异物4种场景进行测试,结果表明:基于YOLOv5n-CND的矿用输送带异物检测算法未出现误检及重复检测的情况,漏检较少,检测框定位准确,对粘连小目标的处理效果更好,能够实现输送带异物的准确检测.
关键词：矿用输送带异物检测粘连小目标检测YOLOv5nC2f模块归一化高斯瓦萨斯坦距离模块Dy Head检测头
分类号：TD634.1(矿山电工)
资助基金：国家重点研发计划(2022YFC3004700)
论文发表日期：2025-01-09
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：7( 38-44 )
英文信息