南京邮电大学学报(自然科学版)

作者：张伟,庄幸涛,王雪力,陈云芳,李延超

单位：南京邮电大学 计算机学院，江苏南京210023南京邮电大学 计算机学院，江苏南京210023南京邮电大学 计算机学院，江苏南京210023南京邮电大学 计算机学院，江苏南京210023南京邮电大学 计算机学院，江苏南京210023

关键词：无人机；小目标检测；密集连接；通道剪枝

基金：国家重点研发计划（2019YFB2101700）资助项目

随着无人机硬件成本的降低和深度学习算法的发展，部署在无人机终端的实时目标检测算法在诸多领域展现出广泛的应用前景。然而，嵌入式设备有限的能耗和算力，以及普适性目标检测算法对于小目标特征提取不够充分等问题，制约了此类算法速度和精度的提升。文中提出了一种部署在无人机终端上的小目标实时检测算法DS YOLO（Dense SPP YOLO），算法基于密集连接的思想设计了全新的主干网络，并改进了空间金字塔池化模块以增强小目标的特征提取和多尺度特征复用，最后基于批归一化层（Batch Normalization）的缩放因子修剪网络中不重要的通道，修剪瘦身后的算法更加适合部署在移动端。在Visdrone2019 DET数据集上的测试结果表明，DS YOLO算法mAP (mean Average Precision)指标比SlimYOLOv3算法提升约3%，检测速度达到89FPS（Frames Per Second），高于SlimYOLOv3的67FPS。

来源：2021年第1期

《南京邮电大学学报(自然科学版)》期刊编辑部

查看南京邮电大学学报(自然科学版)杂志2021年第1期