高精度轻量级的无人机航摄图像目标检测模型研究  

High-Precision and Lightweight Object Detection Model for Drone Aerial Photography Images

作　　者：赵向阳 史再峰 王云峰 牛孝伟 罗韬[2] Zhao Xiangyang;Shi Zaifeng;Wang Yunfeng;Niu Xiaowei;Luo Tao(School of Microelectronics,Tianjin University,Tianjin 300072,China;College of Intelligence and Computing,Tianjin University,Tianjin 300072,China;Tianjin Key Laboratory of Imaging and Sensing Microelectronic Technology,Tianjin 300072,China)

机构地区：[1]天津大学微电子学院,天津300072 [2]天津大学智能与计算学部,天津300072 [3]天津市成像与感知微电子技术重点实验室,天津300072

出　　处：《激光与光电子学进展》2024年第24期412-420,共9页Laser & Optoelectronics Progress

基　　金：国家重点研发计划(2022YFB2804401);国家自然科学基金(62071326)。

摘　　要：针对当前主流的轻量级目标检测模型在无人机航摄场景下检测精度低的问题,提出一种基于YOLOv8s的高精度轻量级航摄图像目标检测模型LEFE-YOLOv8。首先设计内嵌注意力机制的增强特征提取卷积(EFEConv),其与部分通道卷积(PConv)及1×1卷积共同组成轻量级增强特征提取模块,在提高模型特征提取能力的同时降低了参数量与计算量。然后在特征融合网络中引入轻量级动态上采样算子模块,有效地解决了高层特征网络在上采样过程中的信息丢失问题。最后构建带有多尺度模块的检测头,提高了网络模型的多尺度检测能力。最终的实验结果表明,相较于基准模型,所提改进模型在VisDrone2019和HIT-UAV数据集上的平均精度分别达42.3%与83.9%,且参数量在10×10~6以下,更加适合航摄图像目标检测任务。The current mainstream lightweight object detection models exhibit low detection accuracy in unmanned aerial vehicle(UAV)photography scenes.This study introduces a high-precision and lightweight aerial photography image object detection model based on YOLOv8s,named LEFE-YOLOv8.First,an enhanced feature extraction convolution(EFEConv)incorporating an attention mechanism was developed.It is integrated with partial channel convolution(PConv)and 1×1 convolution to create a lightweight enhanced feature extraction module.This integration augments the model's feature extraction capabilities and reduces the number of parameters and computational complexity.Subsequently,a lightweight dynamic upsampling operator module was incorporated into the feature fusion network,effectively addressing the information loss problem during the upsampling process in high-level feature networks.Finally,a detection head with multi-scale modules was designed to enhance the network model's multi-scale detection capabilities.The final experimental results demonstrate that,compared with the benchmark model,the improved model achieves an average accuracy of 42.3%and 83.9%on the VisDrone 2019 and HIT-UAV datasets,respectively,with less than 10×10~6 parameters.These results establish the model's suitability for aerial image object detection tasks.

关 键 词：目标检测 无人机航摄图像 增强特征提取 动态上采样 多尺度模块 

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