-
基于改进Mask R-CNN的矿石类型检测算法
肖成勇a,李擎a,栗辉a,王莉a,陈子一a,张德政b,车伟杰a (北京科技大学 a.自动化学院;b.计算机与通信工程学院,北京 100083) 摘要:针对不同的矿石类型进行选矿工艺参数设置和操作有利于保障精矿质量,提高回收率和降低物资消耗。由于缺乏矿石类型实时检测有效手段,选矿过程的精准操作目前较难实现。为此,本文提出一种全面改进的Mask R-CNN矿石类型检测算法。算法步骤:① 使用ResNetV1d-50提取矿石图像各阶段的特征图,并在主干网络中加入可变形卷积以便增强异形矿石的特征;② 改进FPN,通过在主干网络的C5特征层加入特征残差模块,并融合到P5特征层,得到具有更强语义信息的多尺度特征图;③ 改进RPN,设计自适应的正样本IOU选取方案来匹配宽高比异常的矿石,进一步提高异形矿石的识别精度;④ 在RoIAlign网络基础上加入Global-Context,以提高小矿石的检测能力;⑤ 在数据增强和训练技巧方面对模型进行改进。结果表明,本文算法的平均精度为67.92%,平均交并比为63.54%,分别比基准模型提高了13.67%和9.71%。本文研究方法在矿石类型识别领域具有较好的应用价值。 关键词:Mask R-CNN;矿石类型识别;可变形卷积;训练技巧
(责任编辑:zgltw)