The detection of small-scale targets, characterized by limited available features and unclear textures, has perennially posed a challenge in the field of object detection. To address issues related to false positives and false negatives unmanned aerial vehicle (UAV) targets, we propose an improved UAV small-target detection algorithm, termed LASD-YOLOv5. This algorithm introduces a polarized self-attention mechanism to more accurately extract minute features, incorporates a weighted bidirectional feature pyramid network to replace the path aggregation network, thus enhancing the utilization of low-level features. Furthermore, it decouples the detection head to expedite model convergence. Additionally, to tackle the scarcity of small targets and incomplete scene coverage in existing UAV small-target datasets, we contribute a multi-scene, low-speed, small UAV target dataset (LASD-D).The experimental results demonstrate that our proposed algorithm achieves an average precision of 98.29% in LASD-D, surpassing the baseline network by 2.87%. Notably, it outperforms mainstream algorithms such as YOLOv7、 YOLOv8 and QueryDet, effectively meeting the demands of UAV small-target detection applications.
HADIH J, CAOY, NISAK U, et al. A comprehensive survey on security, privacy issues and emerging defence technologies for UAVs[J]. Journal of Network and Computer Applications, 2023, 213: 103607.
[2]
KRIZHEVSKYA, SUTSKEVERI, HINTONG E. ImageNet classification with deep convolutional neural networks[J]. Communications of the ACM, 2017, 60(6): 84-90.
[3]
REDMONJ, DIVVALAS, GIRSHICKR, et al. You only look once: unified, real-time object detection[C]//IEEE Conference on Computer Vision & Pattern Recognition, 2016: 779-788.
[4]
REDMONJ, FARHADIA. YOLOv3: an incremental improvement[C]//IEEE Conference on Computer Vision and Pattern Recognition, 2017: 6517-6525.
[5]
BOCHKOVSKIYA, WANGC Y, LIAOH Y M. YOLOv4: Optimal speed and accuracy of object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2020.
[6]
LIUW, ANGUELOVD, ERHAND, et al. SSD: single shot MultiBox detector[C]//European Conference on Computer Vision. Cham: Springer, 2016: 21-37.
[7]
LAWH, DENGJ. CornerNet: detecting objects as paired keypoints[J].International Journal of Computer Vision, 2020, 128(3): 642-656.
[8]
YANGC, HUANGZ, WANGN. QueryDet: Cascadedsparse query for accelerating high-resolution small objectdetection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022: 13668-13677.
[9]
GIRSHICKR, DONAHUEJ, DARRELLT, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2014: 580-587.
[10]
GIRSHICKR. Fast R-CNN[C]// 2015 IEEE International Conference on Computer Vision (ICCV), 2015: 1440-1448.
[11]
RENS, HEK, GIRSHICKR, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2017, 39(6): 1137-1149.
WANGJiannan, ShengtaoLÜ, NIUJian. Drone detection method based on improved YOLOv5[J]. Optics & Optoelectronic Technology, 2022, 20(5): 48-56. (in Chinese)
[14]
LUOK, ZHUG, LIY, et al. Low altitude and low speed uav identification based on hybrid model[C]//Proceedings of the 3rd International Conference on Computer Engineering, Information Science & Application Technology (ICCIA 2019), 2019: 76-87.
BAOWenqi, XIELiqiang, XUCaihua, et al. Real time detection method of micro UAV based on YOLOv5[J]. Journal of Ordnance Equipment Engineering, 2022, 43(5): 232-237. (in Chinese)
[17]
WANGM, YANGW, WANGL, et al. FE-YOLOv5: Feature enhancement network based on YOLOv5 for small object detection[J]. Journal of Visual Communication and Image Representation, 2023, 90: 103752.
[18]
ZHUX, LYU S, WANGX, et al. TPH-YOLOv5: Improved yolov5 based on transformer prediction head for object detection on drone-captured scenarios[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision, 2021: 2778-2788.
[19]
LIUH, LIUF, FANX, et al. Polarized self-attention: towards high-quality pixel-wise mapping[J]. Neurocomputing, 2022, 506: 158-167.
[20]
LINT Y, DOLLARP, GIRSHICKR, et al. Feature pyramid networks for object detection [C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2017: 936-944.
[21]
LIUS, QIL, QINH, et al. Path aggregation network for instance segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2018: 8759-8768.
[22]
GHIASIG, LINT Y, LEQ V, et al. NAS-FPN: learning scalable feature pyramid architecture for object detection[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2019: 7029-7038.
[23]
TANM, PANGR, LEQ V. EfficientDet: scalable and efficient object detection[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020: 10778-10787.
[24]
GEZ, LIUS, WANGF, et al. YOLOX: exceeding YOLO series in 2021[EB/OL].
[25]
SUNH, YANGJ, SHENJ, et al. TIB-Net: drone detection network with tiny iterative backbone[J]. IEEE Access, 2020, 8: 130697-130707.