高级检索

煤矿无人驾驶电机车多目标实时检测及嵌入式部署

Real-time multi-object detection and embedded deployment for autonomous locomotives in coal mines

  • 摘要: 煤矿轨道电机车无人驾驶是实现辅助运输智能化的关键环节。然而,井下无人驾驶电机车运行环境复杂,受低光照及强遮挡影响,易出现远距离小目标漏检、人员密集导致特征混淆,以及嵌入式算力受限下精度与实时性难以协同优化等问题。针对上述问题,提出一种兼顾检测精度与部署效率的轻量化算法GSE-YOLO。首先,设计全局上下文卷积(Global Context Convolution, GCConv)作为骨干网络的关键特征层,通过融合局部特征与全局上下文信息,使局部特征聚焦目标细节信息,全局上下文信息捕捉目标全局语义和背景信息,进而增强复杂环境下目标特征的辨识能力;其次,引入空间深度转换卷积(Space-to-Depth Convolution, SPDConv)优化下采样过程,通过在空间维度上进行深度转换,使得细节信息损失降到最低,提升远距离小目标特征表达能力;最后,构建融合通道与空间双重注意力的C2f-EACF模块,强化对目标关键区域的关注度,抑制背景干扰,提高遮挡场景中的特征区分能力与边界定位精度。试验结果表明:GSE-YOLO模型的mAP@0.5和mAP@0.5:0.95分别达到85.2%和48.5%,在参数量仅为2.98×106的条件下较基线模型YOLOv8n分别提升了3.0%和2.4%。结合TensorRT加速策略在Jetson Xavier NX平台上进行测试,FP16量化模式下单帧推理总耗时仅为31.02 ms。该方法在保持模型轻量化的同时,有效提升了复杂真实路况下的检测鲁棒性与实时性,满足煤矿辅运系统边缘端部署的工程需求,为井下电机车无人驾驶技术的实际应用提供技术支撑。

     

    Abstract: Autonomous driving of rail locomotives in coal mines is regarded as a critical link in achieving intelligent auxiliary transportation. However, the operation environment of underground autonomous locomotives is highly complex. Affected by low illumination and severe occlusion, issues such as missed detection of long-distance small targets and feature confusion caused by densely populated personnel are easily triggered. Furthermore, the synergistic optimization of detection accuracy and real-time performance is difficult to achieve under limited embedded computing power. To address the aforementioned issues, a lightweight algorithm named GSE-YOLO, which balances detection accuracy and deployment efficiency, is proposed. First of all, a global context convolution (GCConv) is designed as the key feature layer of the backbone network. By fusing local features and global context information—where local features focus on target detail information, while global context information captures the global semantic and background information of the target—the identification capability of target features in complex environments is enhanced. Secondly, a Space-to-Depth convolution (SPDConv) is introduced to optimize the downsampling process. By performing depth transformation in the spatial dimension, the loss of detailed information is minimized, thereby improving the feature representation capability for long-distance small targets. Finally, a C2f-EACF module integrating dual channel and spatial attention is constructed to strengthen the attention towards key target regions and suppress background interference. Consequently, the feature discrimination capability and bounding box localization accuracy in occluded scenes are prominently improved. Experimental results demonstrate that the mAP@0.5 and mAP@0.5:0.95 of the GSE-YOLO model reach 85.2% and 48.5%, respectively. Under the condition of a parameter scale of only 2.98×106, the metrics are improved by 3.0% and 2.4%, respectively, compared with the baseline model YOLOv8n. Furthermore, when tested on the Jetson Xavier NX platform combined with a TensorRT acceleration strategy, the total inference time per frame under the FP16 quantization mode is contained at merely 31.02 ms. While maintaining the lightweight nature of the model, the detection robustness and real-time performance under complex real-world conditions are effectively enhanced, meeting the engineering requirements for edge deployment in underground auxiliary transportation systems, and providing technical support for the practical application of autonomous driving technology for underground locomotives.

     

/

返回文章
返回