Cross-scale feature alignment and feature enhancement for small object detection
摘要
Small object detection has always been a key challenge in the field of object detection due to its low resolution and limited information. Traditional object detection algorithms often struggle with small objects, as they fail to fully leverage low-level features and overlook the semantic gaps between feature maps. To address these issues, we introduce a novel small object detector called SO-YOLO. Our approach incorporates a cross-scale feature alignment (CSFA) module that employs high-level semantics to guide the spatial redistribution of low-level features, thereby reducing semantic gaps during feature fusion. Additionally, a small object feature enhancement (SOFE) module is designed to extract relevant spatial features from low-level feature maps and filter out background noise. Subsequently, a high-resolution feature pyramid network (HRFPN) is utilized to upsample all feature maps to the same high resolution, thereby increasing the likelihood of detecting small objects. Extensive experiments conducted on the challenging VisDrone2019, TT-100 K, and SeaDronesSee datasets demonstrate that SO-YOLO significantly improves the detection performance of small objects. Code is available at https://github.com/Anewman211/SO-YOLO.