金属矿山

北大核心,CA,JST,WJCI,

国内刊号:34-1055/TD

国际刊号:1001-1250

金属矿山杂志2024年第11期:基于注意力增强YOLOv5l 的矿粉品位识别算法 优化研究

发布日期:

作者:丁鹏益,徐振洋,郭连军,王雪松

单位:1. 辽宁科技大学矿业工程学院,辽宁 鞍山 114051;2. 辽宁省金属矿产资源绿色开采工程研究中心,辽宁 鞍山114051;3. 沈阳工业大学建筑与土木工程学院,辽宁 沈阳 110870

关键词:铁矿,矿石品位,YOLOv5,图像识别,卷积注意力模块,挤压和激发模块,

现阶段传统化学分析方法获取品位存在费时费力等问题,通过图像识别分析块状矿石品位又存在形状干扰严重的问题,为此提出了一种基于YOLOv5 针对矿石矿粉特征的图像识别方法。同时,在训练过程中添加卷积注意力模块(Convolutional Block Attention Module,CBAM)、挤压和激发模块(Squeeze and Excitation Module,SENet)进入训练网络,通过注意力机制增强学习矿粉具体特征的能力,将注意力聚焦于矿粉明显细节,忽略矿粉图中无用信息,提高识别精度;其次通过修改损失函数增强其分类效果,研究损失函数对矿粉识别效果的影响。研究表明:在铁矿粉品位识别中,添加CBAM 注意力模块的网络模型识别矿粉的训练精度达到86%,使用SENet 注意力模块的网络训练精度为80%,均略高于原有模型的79%,修改损失函数的网络模型训练精度降低了5%,得出YOLOv5l+CBAM 且损失函数设置为0. 5 的网络模型最佳。研究结果反映出所提方法对矿粉特征图像识别具有一定的适用性。

来源:2024年第11期

《金属矿山》期刊编辑部

查看金属矿山杂志2024年第11期

联系我们

  • 地址:安徽省马鞍山市经济开发区西塘路666号
  • 电话:0555-2404809
  • E-mail:jsks@vip.163.com

咨询工作人员