融合CBAM的YOLOv4轻量化检测方法  

YOLOv4 Lightweight Detection Method Based on CBAM

作　　者：任丰仪 裴信彪 乔正 白越[1] REN Feng-yi;PEI Xin-biao;QIAO Zheng;BAI Yue(Changchun Institute of Optics,Fine Mechanics and Physics,Chinese Academy of Sciences,Changchun 130033,China;University of Chinese Academy of Sciences,Beijing 100039,China)

机构地区：[1]中国科学院长春光学精密机械与物理研究所,长春130033 [2]中国科学院大学,北京100039

出　　处：《小型微型计算机系统》2023年第5期1008-1014,共7页Journal of Chinese Computer Systems

基　　金：国家自然科学基金项目(11372309,61304017)资助;吉林省科技发展计划重点项目(20150204074GX,20160204010NY)资助;省院合作科技专项资金项目(2020SYHZ0031)资助;中科院轻型动力创新院重点基金项目(CXYJJ20-ZD-03)资助;中科院青促会项目(2014192)资助。

摘　　要：基于深度学习的目标检测算法应用于无人机视觉中,会极大提升无人机的场景理解能力,但模型参数量和计算量巨大,难以应用于移动端或嵌入式平台.因此本文提出了一种效果较好的轻量级实时检测模型,采用YOLOv4模型网络作为主要参考模型,使用MobileNet替换主干网络,并通过添加CBAM注意力机制以及Soft-NMS后处理策略来提高模型的准确性.选用PASCAL VOC数据集来测试所提出的轻量级YOLOv4模型,结果显示参数量只有原模型的一半,但速度FPS提升了26.48,精度mAP只下降了0.52%.将所提出的轻量化YOLOv4模型部署Nvidia Jetson TX2低功耗系统以及树莓派上,飞行试验显示在TX2上模型FPS达到了21.8,是原始的YOLOv4的4.74倍,将本算法部署到无人机装载的嵌入式平台上,能够对航拍视野中的车辆目标进行实时识别和定位.Object classification and detection algorithms based on deep learning will greatly improve understanding capabilities when applied to UAV vision,but the amount of model parameters and calculations are too huge to apply to mobile or embedded platforms.This paper investigated the feasibility and accuracy of a new lightweight real-time detection model with main reference model used the YOLOv4 model network andMobileNet replaced the backbone network,as well as CBAM attention mechanism and Soft-NMS post-processing strategy.The PASCAL VOC data set was utilized to test the proposed lightweight YOLOv4 model.The results showed that the parameter amount was only half of the original model and the speed FPS was increased by 26.48 justwith 0.52%reduced mAP.After the model is deployed on the Nvidia Jetson TX2 low-power system,its FPS has reached 21.8,which is 4.74 times that of the original YOLOv4.The algorithm is applied to the embedded platform of the drone,which benefits to real-time recognition and positioning of vehicle targets in the aerial view.

关 键 词：无人机图像 YOLOv4 MobileNet CBAM 柔性非极大抑制策略 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]