SOO-YOLO: an efficient small object detection model for UAV images
摘要
Object detection in UAV images is a challenging task in computer vision. Objects in UAV images have difficulties such as low resolution, large-scale variations, and complex backgrounds. In this study, we propose a novel real-time detection model based on YOLO11 to solve the above problems. First, we created a better feature extraction module to better extract the small-object features. Second, we designed a feature pyramid that improves the model detection by using local and global information of objects. In addition, we design a task interaction detection header that improves model localization and classification through task alignment. We also developed Inner-Wise-MPDIoUv2 to address the limitations of CIoU in detecting small objects. Finally, we use model pruning to reduce the model size. SOO-YOLO achieves a 6.1% improvement over YOLO11n on the VisDrone2019-DET dataset’s