Real-Time Small Drones Detection Based on Pruned YOLOv4.
Real-Time Small Drones Detection Based on Pruned YOLOv4.
复制标题
基于剪枝YOLOv4的实时小型无人机检测
DOI:
10.3390/s21103374
复制
发表时间:
2021-05-12
期刊:
影响因子:
--
通讯作者:
Li N
中科院分区:
文献类型:
--
作者:
Liu H;Fan K;Ouyang Q;Li N
To address the threat of drones intruding into high-security areas, the real-time detection of drones is urgently required to protect these areas. There are two main difficulties in real-time detection of drones. One of them is that the drones move quickly, which leads to requiring faster detectors. Another problem is that small drones are difficult to detect. In this paper, firstly, we achieve high detection accuracy by evaluating three state-of-the-art object detection methods: RetinaNet, FCOS, YOLOv3 and YOLOv4. Then, to address the first problem, we prune the convolutional channel and shortcut layer of YOLOv4 to develop thinner and shallower models. Furthermore, to improve the accuracy of small drone detection, we implement a special augmentation for small object detection by copying and pasting small drones. Experimental results verify that compared to YOLOv4, our pruned-YOLOv4 model, with 0.8 channel prune rate and 24 layers prune, achieves 90.5% mAP and its processing speed is increased by 60.4%. Additionally, after small object augmentation, the precision and recall of the pruned-YOLOv4 almost increases by 22.8% and 12.7%, respectively. Experiment results verify that our pruned-YOLOv4 is an effective and accurate approach for drone detection.
登录
查看更多内容
影响因子:
11.2
作者:
Shi, Xiufang;Yang, Chaoqun;Chen, Jiming
通讯作者:
Chen, Jiming
影响因子:
6.8
作者:
Anwar, Muhammad Zohaib;Kaleem, Zeeshan;Jamalipour, Abbas
通讯作者:
Jamalipour, Abbas
DOI:
10.1109/tpami.2016.2572683
发表时间:
2017-04-01
影响因子:
23.6
作者:
Shelhamer, Evan;Long, Jonathan;Darrell, Trevor
通讯作者:
Darrell, Trevor
影响因子:
5
作者:
Nowak, Aleksander;Naus, Krzysztof;Maksimiuk, Dariusz
通讯作者:
Maksimiuk, Dariusz
影响因子:
5
作者:
Koga, Yohei;Miyazaki, Hiroyuki;Shibasaki, Ryosuke
通讯作者:
Shibasaki, Ryosuke