高级检索

基于CDM-RTDETR的煤矿人员跟踪计数方法

Coal mine personnel tracking and counting method based on CDM-RTDETR

  • 摘要: 煤矿安全生产中实时监测人员数量及空间分布是保障矿工生命安全、预防事故发生的关键,然而当前基于计算机视觉的人员计数技术在复杂煤矿环境下仍面临诸多挑战。针对在复杂煤矿环境中因多尺度变化现象和非人员目标干扰,现有方法在多尺度特征融合及聚焦能力上存在不足,导致误检、漏检和计数不准确的问题,提出了一种基于CDM-RTDETR的煤矿人员跟踪计数方法。该方法首先对RTDETR检测模型进行优化:在主干网络中采用PSConv优化C2f模块,构建主干网络CSP-AP以增强特征提取能力;在编码器部分将可变形注意力机制与尺度内特征交互模块相结合,设计基于可变形注意力的尺度内特征交互DAIFI以提升对人员关键特征的聚焦能力;并提出多尺度特征聚合扩散金字塔网络MADPN以优化多尺度特征的融合与传递。然后在优化检测模型的基础上,结合ByteTrack目标跟踪算法实现人员轨迹的稳定跟踪与准确计数。最后在构建的煤矿人员检测数据集和煤矿人员跟踪数据集上对所述方法进行实验验证。结果表明:检测性能方面,相比于原RTDETR检测模型,CDM-RTDETR的准确率和召回率分别提升了2.3%和3.8%,参数量减少了12.4%,平均检测精度mAP@0.5达到97.8%;跟踪性能方面,跟踪准确率MOTA达到了90.8%,跟踪精确度MOTP达到了84.1%;计数准确度方面,人员计数综合准确率达到95.3%。

     

    Abstract: In coal mine safety production, real-time monitoring of the number and spatial distribution of personnel is a key basis for ensuring miners’ life safety and preventing accidents. However, current computer vision-based personnel counting technology still faces numerous challenges in complex coal mine environments. Given the problems of false detection, missed detection, and inaccurate counting, which are caused by insufficient multi-scale feature fusion and focusing capabilities of existing methods that result from multi-scale variation phenomena and non-personnel target interference in complex coal mine environments, a coal mine personnel tracking and counting method based on CDM-RTDETR is proposed. The RTDETR detection model is optimized through three aspects: within the backbone network, the C2f module is optimized using PSConv, and the backbone network CSP-AP is constructed to enhance feature extraction capability; in the encoder, the deformable attention mechanism is combined with the intra-scale feature interaction module to design the Deformable Attention-based Intra-scale Feature Interaction(DAIFI) to improve the focusing capability on the key features of personnel; additionally, the Multi-scale Feature Aggregation and Diffusion Pyramid Network(MADPN) is proposed to optimize the fusion and transfer of multi-scale features. On the basis of the optimized detection model, the ByteTrack target tracking algorithm is incorporated to achieve stable tracking of personnel trajectories and accurate counting. Experimental verification is conducted on the constructed coal mine personnel detection dataset and coal mine personnel tracking dataset. Results show that in terms of detection performance: compared with the original RTDETR detection model, the precision and recall are improved by 2.3% and 3.8% respectively, the number of parameters is reduced by 12.4%, and the mean average precision (mAP@0.5) reaches 97.8%. In terms of tracking performance, the Multiple Object Tracking Accuracy(MOTA) reaches 90.8%, and the Multiple Object Tracking Precision(MOTP) reaches 84.1%. In terms of counting accuracy, the comprehensive accuracy of personnel counting reaches 95.3%.

     

/

返回文章
返回