国内刊号:43-1215/TD
国际刊号:1005-2763
发布日期:
作者:高凯,董立红,邓凡
关键词:带 式 运 输;煤 块 检 测;目 标 检 测;注 意 力 机 制;YOLOv5
针对煤矿带式输送机上煤块检测因光照不均存在的漏 检与误检问题,提出一种基于门控卷积和上下文注意力机制 的改进 YOLOv5煤块检测算法.首先,将主干网络中残差 模块替换为递归门控卷积模块,通过逐步融合特征信息,提 取高阶语义特征,增强模型的特征提取能力.其次,在特征 融合结构中加入 GCA 注意力机制,将全局上下文信息融入 坐标注意力模块中,加强感兴趣区域的全局表示,增强多尺 度特征融合能力,提高模型对煤块边缘特征的敏感度.最 后,采用SIoU 损失函数,加速网络模型的收敛.试验结果 表明,改进的算法在自建煤块数据集上平均精度均值达到92.8%,召回率达到85.9%,检测速度达到38帧/s.既提高 了检测精度,又满足了检测的实时性.
来源:2023年第6期
《矿业研究与开发》期刊编辑部