Laser & Optoelectronics Progress, Volume. 60, Issue 6, 0610004(2023)
Small-Target Detection Network Based on Adaptive Feature Enhancement
Small-target detection from images remains a challenge in the field of computer vision because of the limited size, small appearance and geometric clues, and lack of large-scale small-target datasets. To solve this issue, an adaptive feature-enhanced target detection network called YOLO-AFENet is proposed to improve the accuracy of small-target detection. First, by introducing the feature fusion factor, an improved adaptive bidirectional feature fusion module is designed using feature maps of various scales to improve the network's feature expression ability. Second, combined with the network characteristics, a spatial attention generation module is proposed to improve the network's feature localization ability by identifying the location information of the region of interest in the image. The experimental results of the UAVDT dataset show that YOLO-AFENet has a 6.3 percentage points higher average accuracy compared with YOLOv5 and is better than other target-detection networks.
Get Citation
Copy Citation Text
Mengmeng Wu, Zebin Zhang, Yaozhe Song, Ziting Shu, Baoqing Li. Small-Target Detection Network Based on Adaptive Feature Enhancement[J]. Laser & Optoelectronics Progress, 2023, 60(6): 0610004
Category: Image Processing
Received: Nov. 24, 2021
Accepted: Jan. 14, 2022
Published Online: Mar. 16, 2023
The Author Email: Li Baoqing (sinoiot@mail.sim.ac.cn)