基于Faster-RCNN 与自注意力机制的矿山图像
异常检测算法

金属矿山 ›› 2024, Vol. 53 ›› Issue (7): 196-.

基于Faster-RCNN 与自注意力机制的矿山图像异常检测算法

张玉茜1　刘文荣1　孙　勇2　刘丰武1　殷齐月1　马文宁1　赵建伟1

1. 山东省地质矿产勘查开发局第一地质大队,山东济南 250014;2. 山东建筑大学测绘地理信息学院,山东济南 250101

出版日期:2024-07-15 发布日期:2024-09-01

Mining Image Anomaly Detection Algorithm Based on Faster-RCNN and Self-attention Mechanism

ZHANG Yuqian1　LIU Wenrong1　SUN Yong2　LIU Fengwu1　YIN Qiyue1　MA Wenning1　ZHAO Jianwei1

Online:2024-07-15 Published:2024-09-01

摘要/Abstract

摘要： 矿山异常行为检测是一项重要工作,有助于提高采场安全生产监管效率。提出了一种基于自注意力机制的 Faster-RCNN 算法,用于矿山图像异常行为检测。该算法通过自注意力机制对特征图进行加权,有效提取关键特征,并减少冗余信息干扰。首先,从图像中提取RoI 区域,并通过自注意力机制对RoI 区域内的特征图进行加权,使得关键特征得到更准确的提取。同时,对RoI 区域内的特征图进行细粒度融合,以捕捉更多的图像细节信息。最后,使用分类器对每个RoI 区域进行分类,以检测图像中的异常行为。在自建数据集上对算法进行了训练与评估,试验结果表明:所提算法在矿工图像异常行为检测工作上表现出更高的准确性和鲁棒性,与传统的Faster-RCNN 算法相比,该算法检测精度提高了4. 8%。此外,该算法对于光照和角度等变化具有更好的鲁棒性,可以有效应对实际场景中的复杂环境。

张玉茜, 刘文荣, 孙　勇, 刘丰武, 殷齐月, 马文宁, 赵建伟. 基于Faster-RCNN 与自注意力机制的矿山图像异常检测算法[J]. 金属矿山, 2024, 53(7): 196-.

ZHANG Yuqian, LIU Wenrong, SUN Yong, LIU Fengwu, YIN Qiyue, MA Wenning, ZHAO Jianwei. Mining Image Anomaly Detection Algorithm Based on Faster-RCNN and Self-attention Mechanism[J]. Metal Mine, 2024, 53(7): 196-.

扫码分享