高级检索

改进的YOLOv8n轻量化火星表面岩石检测算法

Lightweight Mars Surface Rock Detection Algorithm Based on Improved YOLOv8n

  • 摘要: 针对火星探测器在复杂地形中自主导航的安全避障需求及星载平台计算资源与能源供应的双重约束,构建YOLOv8-LMD轻量化检测模型,旨在实现火星表面岩石检测算法需兼具的高精度与轻量化特性要求。基于HGNetv2架构重构轻量化主干网络,实现模型参数的初步压缩;设计了一种多尺度特征融合网络结构,通过集成Slim-neck与ASF-YOLO对颈部网络进行重构,有效提升对不同尺度岩石目标的特征表征能力;采用卷积共享策略设计了轻量级检测头,在降低计算复杂度的同时增强分类定位精度;使用剪枝算法针对模型参数冗余进行修剪,使模型进一步压缩,并通过知识蒸馏技术实现精度的补偿优化。通过实验发现,与YOLOv8n相比,YOLOv8-LMD精度提升1.7%,计算量减少68%,参数量减少77%,模型大小减小75%。因此,可认为本文模型更适合应用于火星表面岩石检测任务。

     

    Abstract: Aiming at the demand for safe obstacle avoidance in the autonomous navigation of Mars rover in complex terrain and the double constraints of computational resources and energy supply of the onboard platform, this paper constructs the YOLOv8-LMD lightweight detection model, aiming at realizing the requirements of high precision and lightweight characteristics of the rock detection algorithm on the surface of Mars. First, the lightweight backbone network is reconstructed based on the HGNetv2 architecture to realize the preliminary compression of model parameters. Secondly, a multi-scale feature fusion network structure is designed, and the neck network is reconstructed by integrating Slim-neck and ASF-YOLO to effectively improve the feature characterization of rock targets at different scales. In addition, a lightweight detection head is designed by using the convolutional sharing strategy, which reduces the computational complexity and enhances the classification and localization accuracy at the same time. Finally, a pruning algorithm is used to prune the model parameter redundancy to further compress the model, and the knowledge distillation technique is used to achieve the compensation and optimization of the accuracy. Through experiments, it is found that compared with YOLOv8n, YOLOv8-LMD accuracy is improved by 1.7%, the computational amount is reduced by 68%, the parameter amount is reduced by 77%, and the model size is reduced by 75%. Therefore, it can be considered that the model in this paper is more suitable to be applied in the Mars surface rock detection task.

     

/

返回文章
返回