期刊
SENSORS
卷 21, 期 10, 页码 -出版社
MDPI
DOI: 10.3390/s21103374
关键词
anti-drone; YOLOv4; pruned deep neural network; small object augmentation
资金
- National Natural Science Foundation of China [61763018]
- 5G Program of Jiangxi Province [20193ABC03A058]
- Education Department of Jiangxi Province [GJJ170493, GJJ190451]
- Program of Qingjiang Excellent Young Talents, Jiangxi University of Science and Technology
- Special Project of Jiangxi Province [20193ABC03A058]
To address the threat of drones intruding into high-security areas, a pruned-YOLOv4 model is proposed in this paper, which reduces model size and increases speed by pruning convolutional channels and shortcut layers, while improving the accuracy of small drone detection through special data augmentation methods.
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.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据