DS-YOLO:一种部署在无人机终端上的小目标实时检测算法

被引：17

作者：

张伟

庄幸涛

王雪力

陈云芳

李延超

机构：

[1] 南京邮电大学计算机学院

来源：

南京邮电大学学报(自然科学版) | 2021年 / 41卷 / 01期

基金：

国家重点研发计划;

关键词：

无人机; 小目标检测; 密集连接; 通道剪枝;

D O I：

10.14132/j.cnki.1673-5439.2021.01.011

中图分类号：

TP391.41 []; V279 [无人驾驶飞机];

学科分类号：

080203 ; 1111 ;

摘要：

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

引用

页码：86 / 98

页数：13

共 6 条

[1] UAV-YOLO: Small Object Detection on Unmanned Aerial Vehicle Perspective[J] . Mingjie Liu,Xianhao Wang,Anjian Zhou,Xiuyuan Fu,Yiwei Ma,Changhao Piao. Sensors . 2020 (8)
[2] Dilated Convolution and Feature Fusion SSD Network for Small Object Detection in Remote Sensing Images[J] . Junsuo Qu,Chang Su,Zhiwei Zhang,Abolfazl Razi. IEEE Access . 2020
[3] Tinier-YOLO: A Real-Time Object Detection Method for Constrained Environments[J] . Fang Wei,Wang Lin,Ren Peiming. IEEE Access . 2020
[4] Automatic Change Detection System over Unmanned Aerial Vehicle Video Sequences Based on Convolutional Neural Networks
Garcia Rubio, Victor
Rodrigo Ferran, Juan Antonio
Menendez Garcia, Jose Manuel
Sanchez Almodovar, Nuria
Lalueza Mayordomo, Jose Maria
Alvarez, Federico
[J]. SENSORS, 2019, 19 (20)
[5] Small Object Detection in Unmanned Aerial Vehicle Images Using Feature Fusion and Scaling-Based Single Shot Detector with Spatial Context Analysis[J] . Liang Xi,Zhang Jing,Zhuo Li,Li Yuzhao,Tian Qi. IEEE Transactions on Circuits and Systems for Video Technology . 2019 (6)
[6] LightDenseYOLO: A Fast and Accurate Marker Tracker for Autonomous UAV Landing by Visible Light Camera Sensor on Drone.[J] . Nguyen Phong Ha,Arsalan Muhammad,Koo Ja Hyung,Naqvi Rizwan Ali,Truong Noi Quang,Park Kang Ryoung. Sensors (Basel, Switzerland) . 2018 (6)

← 1 →