基于YOLOv8的掘进机零部件轻量化识别模型

A Lightweight Recognition Model for Tunneling Machine Components

  • 摘要: 针对混合现实辅助维修中的目标识别无法兼顾准确性与实时性的问题,本文提出一种基于YOLOv8的轻量化识别模型。首先,引入异构卷积模块改进原有的C2f模块,降低浮点运算次数,提高识别速度;其次,使用线性可变形卷积模块替换原主干网络中的部分普通卷积模块,进一步提高网络模型的实时性;然后,采用重参数广义特征金字塔网络重构颈部网络增强特征之间的相互作用,提高模型的特征融合能力;最后,采用更注重目标形状的Shape-IoU损失函数改进原损失函数,使得边界框回归更加准确。在自制的掘进机零部件数据集上的测试结果表明,改进后的模型精确度提高了0.4%,参数量减少了30%,帧率提高了13.9%,在保证准确率的同时提高了识别速度,能够满足混合现实辅助维修矿用设备系统的应用要求。

     

    Abstract: To address the challenge of balancing accuracy and real-time performance in target recognition for Mixed Reality (MR)-assisted maintenance, this paper proposes a lightweight recognition model based on YOLOv8. Firstly, a heterogeneous convolution module is introduced to replace the original C2f module, reducing the number of floating-point operations (FLOPs) and improving recognition speed. Secondly, linear deformable convolution modules are used to replace some standard convolution modules in the original backbone network, further enhancing the model's real-time capability. Thirdly, the neck network is reconstructed using a Reparameterized Generalized Feature Pyramid Network (RepGFPN) to strengthen feature interactions and improve the model's feature fusion ability. Finally, the original loss function is improved by adopting the Shape-IoU loss function, which places greater emphasis on target shape, leading to more accurate bounding box regression. Experimental results on a self-built dataset of tunneling machine components demonstrate that the improved model achieves a 0.4% increase in precision, a 30% reduction in parameter count, and a 13.9% increase in frame rate. This ensures accuracy while significantly improving recognition speed, meeting the application requirements for MR-assisted maintenance systems in mining equipment.

     

/

返回文章
返回