基于锚点定位和跨层修正的输送带跑偏识别算法

    Conveyor belt deviation identification algorithm based on anchor point positioning and cross-layer correction

    • 摘要: 煤矿输送带的运行状态对于安全生产至关重要,而输送带跑偏极大影响输送带的运输效率和安全性,输送带边缘检测是判定输送带是否发生跑偏的重要依据。为了解决传统卷积神经网络(CNN)方法在输送带边缘线检测中因很难构建像素间长距离依赖关系而导致信息丢失以及提取输送带边缘线不准确的问题,提出了基于锚点定位和跨层修正的DETR(Detection Transformer)编解码器网络结构输送带跑偏识别算法。首先定义输送带边缘线锚点并将锚点位置信息分别嵌入自注意力模块和交叉注意力模块,以捕捉更全面的内部依赖关系和获得更精确的图像特征信息,从而提升解码器对输送带边缘线的感知能力;其次在模型训练阶段加入跨层修正策略来为训练的不同阶段赋予不同的监督权重,以加大后期训练对整体阶段的影响,并增加后期阶段对于前期修正的可见性,以减轻中间阶段级联错误造成的潜在影响并提升输送带边缘线的检测准确率;再者通过先验感兴趣区域(ROI)判定来判断输送带是否跑偏。为了有效训练和评估模型性能,采集3 268张煤矿井下输送带各个场景的图片进行训练和测试,与UNet、DeepLab模型、DETR原始网络模型进行了对比验证。验证结果表明:提出的基于锚点定位和跨层修正的DETR编解码器网络结构输送带跑偏识别算法检测输送带边缘线的准确率分别提升了11.15%、9.33%、4.73%,准确率达到97%,检测速度38帧/s,可实现实时检测。

       

      Abstract: The running state of coal mine conveyor belts is of crucial importance for safe production, and conveyor belt deviation has a significant impact on the transportation efficiency and safety of conveyor belts. Conveyor belt edge detection is an important basis for determining whether conveyor belts deviate or not. To address the problems of information loss and inaccurate extraction of conveyor belt edge lines in the traditional convolutional neural network (CNN) methods for conveyor belt edge line detection due to the difficulty in constructing long-distance dependency relationships between pixels, a conveyor belt deviation recognition algorithm based on the DETR (Detection Transformer) encoder-decoder network structure with anchor point positioning and cross-layer correction is proposed. Firstly, the anchor points of the conveyor belt edge lines are defined, and the position information of these anchor points is embedded into the self-attention module and the cross-attention module respectively. This is done to capture more comprehensive internal dependency relationships and obtain more precise image feature information, thereby enhancing the decoder’s perception ability of the conveyor belt edge lines. Secondly, a cross-layer correction strategy is added during the training stage of the model. Different supervision weights are assigned to different training stages so as to increase the influence of the later training on the whole stage and enhance the visibility of the later stage for the correction of the previous stage. In this way, the potential impact caused by cascading errors in the intermediate stage can be reduced, and the detection accuracy of the conveyor belt edge lines can be improved. Thirdly, the prior region of interest (ROI) determination is used to judge whether the conveyor belt has deviated. Finally, in order to effectively train and evaluate the model performance, 3 268 pictures of various scenes of coal mine underground conveyor belts were collected for training and testing. Through comparison and verification with the UNet, DeepLab models and the original DETR network model, the accuracy of detecting the conveyor belt edge lines by the proposed conveyor belt deviation recognition algorithm based on the DETR encoder-decoder network structure with anchor point positioning and cross-layer correction was increased by 11.15%, 9.33% and 4.73% respectively. By comparing with the on-site actually measured conveyor belt misalignment data, the conveyor belt deviation recognition accuracy correction reached 97%, and the detection speed reached 38 FPS, enabling real-time detection.

       

    /

    返回文章
    返回