高级检索

改进的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: Due to 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, a lightweight detection model, YOLOv8-LMD, was constructed, 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 was reconstructed based on the HGNetv2 architecture to realize the preliminary compression of model parameters. Secondly, a multi-scale feature fusion network structure was designed, and the neck network was 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 was designed by using the convolutional sharing strategy, which reduced the computational complexity and enhanced the classification and localization accuracy at the same time. Finally, a pruning algorithm was used to prune the model parameter redundancy to further compress the model, and the knowledge distillation technique was used to achieve the compensation and optimization of the accuracy. Through experiments, it is found that compared with YOLOv8n, YOLOv8-LMD accuracy was improved by 1.7%, the computational amount was reduced by 68%, the parameter amount was reduced by 77%, and the model size was reduced by 75%. Therefore, it can be concluded that the model proposed in this paper is more suitable for the task of rock detection on the surface of Mars.

     

/

返回文章
返回