高级检索

露天矿山下无人矿卡的轻量级障碍检测算法

Lightweight detection algorithms for small targets on unmanned mining trucks

  • 摘要: 随着人工智能技术不断发展和智慧矿山理念的逐步推进,传统露天矿山的运营模式正在被自动化方式取代。无人矿卡作为智慧矿山的重要组成部分,其推广应用有效解决了因矿区地形不规则、路面坑洼或恶劣天气等因素导致的矿卡翻车、侧滑等问题,从而显著降低了由此引发的伤亡事故。准确的检测目标类别是做避障决策的前提,而模型轻量化可以在资源有限的条件下很好地部署。因此,针对露天矿山场景下,无人矿卡目标检测算法存在参数量多、模型较大及小目标和遮挡目标检测准确率低的问题,提出轻量级无人矿卡检测算法LWHP(Lightweight High-Precision),设计思路有以下4点:其一,提出高效加权双向的特征金字塔网络R-BiFPN,利用这一结构重构颈部网络,通过跨层连接及双向传播,减少冗余计算路径,并通过加权特征融合方式增强多尺度特征融合能力,提升小目标检测能力的同时大幅度降低参数量;其二,设计带有多头注意力机制的检测解耦头,改善卷积层冗余导致网络复杂的问题,并处理空间维度以集中捕捉目标特征,减弱无关背景干扰,提升遮挡目标识别准确率;其三,利用双重卷积构建轻量级神经网络CDC,增强通道间信息流动,提高模型特征表达能力并降低模型复杂度;其四,引入EIOU损失函数,分别计算目标边界框的宽高差异,并加入Focal Loss解决难易样本不平衡问题,获得更快的收敛速度和更优秀的定位能力。经试验表明,改进后算法相较于原始算法参数量降低50.2%,计算量减少46.3%,模型大小压缩47.6%,仅有3.3 MB,且FPS达到92.9,满足实时性需求。精度提升1.6%,召回率提升3.1%,平均精度达到79.6%,相比原模型提升2%,保证轻量级部署的同时提升了检测准确率。

     

    Abstract: With the continuous development of artificial intelligence technology and the gradual advancement of the smart mining concept, traditional open-pit mining operations are being replaced by automated methods. Unmanned mining trucks are a crucial component of smart mining. Their widespread adoption has effectively addressed safety issues caused by irregular terrain, uneven road surfaces, and adverse weather conditions, such as truck overturning and skidding, thereby significantly reducing the occurrence of accidents and fatalities.Accurate target category detection is a prerequisite for obstacle avoidance decisions, and model lightweighting facilitates deployment in resource-limited conditions. Therefore, to address the issues of high parameter count, large model size, and low detection accuracy for small and occluded targets in open-pit mining scenarios, we propose the Lightweight Unmanned Mining Truck Detection Algorithm LWHP (Lightweight High-Precision). This algorithm introduces the Efficient Weighted Bidirectional Feature Pyramid Network (R-BiFPN), which reconstructs the neck network through cross-layer connections and bidirectional propagation, reducing redundant computation paths. It enhances multi-scale feature fusion capability via weighted feature fusion, significantly reducing parameter count while improving small target detection capability. Additionally, it designs a detection decoupling head with a multi-head attention mechanism to improve the issue of network complexity caused by convolutional layer redundancy, processes spatial dimensions to focus on capturing target features, reduces interference from irrelevant backgrounds, and enhances the accuracy of occluded target recognition.urthermore, it constructs a lightweight neural network with dual convolution (CDC), enhancing inter-channel information flow, improving model feature expression capability, and reducing model complexity. The introduction of the Focal-EIOU loss function calculates the width and height differences of target bounding boxes and uses Focal Loss to address the imbalance of difficult and easy samples, achieving faster convergence and superior localization capability. Experiments show that the improved algorithm reduces parameters by 50.2%, computational load by 46.3%, and model size by 47.6% compared to the original algorithm, with a model size of only 3.3 MB. And the FPS reaches 92.9, meeting the real-time requirements.Accuracy is increased by 1.6%, recall by 3.1%, and average precision reaches 79.6%, a 2% improvement over the original model, ensuring lightweight deployment while enhancing detection accuracy.

     

/

返回文章
返回