摘要
针对目标检测算法部署在边缘设备的轻量化需求,提出一种基于改进YOLOv10的轻量化目标检测算法(CMD-YOLO算法)。该算法利用跨尺度特征融合模块对YOLOv10算法的网络结构进行改进,减少了算法模型的参数量与计算量;采用基于Mamba的线性注意力机制改进的部分自注意力模块替换传统的部分自注意力模块,进一步降低了算法模型的参数量;利用空间深度转换卷积模块替换部分传统卷积模块,增强了算法模型对下采样细节信息的提取能力;利用动态上采样器DySample替换传统的上采样模块,在保持上采样精度的同时,降低了算法模型的计算延迟。实验结果表明,CMD-YOLO算法与YOLOv10-n算法相比,在检测精度略微提升的同时,算法模型参数量降低了30.5%,计算量下降了19%,权重文件缩小了29.3%,计算延迟减少了8.8%,能够满足目标检测算法部署在边缘设备中的轻量化需求。
Aiming at the lightweight requirements of deploying object detection algorithms on edge devices,a lightweight object detection algorithm based on improved YOLOv10(CMD-YOLO algorithm)is proposed.This algorithm utilizes a cross-scale feature fusion module to improve the network structure of YOLOv10 algorithm,reducing the parameter and computational complexity of the algorithm model;Adopting a Mamba based linear attention mechanism to improve the partial self attention module and replace the traditional partial self attention module,further reducing the parameter count of the algorithm model;Replacing some traditional convolution modules with spatial depth conversion convolution modules enhances the algorithm model's ability to extract downsampling detail information;By using the dynamic UpSampler DySample to replace the traditional upsampling module,the computational delay of the algorithm model is reduced while maintaining upsampling accuracy.The experimental results show that compared with the YOLOv10-n algorithm,the CMD-YOLO algorithm has slightly improved detection accuracy,reduced model parameters by 30.5%,decreased computational complexity by 19%,reduced weight files by 29.3%,and reduced computational latency by 8.8%,which can meet the lightweight requirements of object detection algorithm deployment in edge devices.
作者
刘印
龚长友
徐国栋
LIU Yin;GONG Changyou;XU Guodong(Southwest Forestry University,Kunming 650224,China;Bingtuan Xingxin Vocational and Technical College,Tiemenguan 841007,China)
出处
《自动化与信息工程》
2025年第1期29-35,共7页
Automation & Information Engineering
