YOLOv8s-DroneNet: Small Object Detection Algorithm Based on Feature Selection and ISIoU
Source abstract
Abstract
Object detection plays a critical role in drone imagery analysis, especially in remote sensing applications where accurate and efficient detection of small objects is essential. Despite significant advancements in drone imagery detection, most models still struggle with small object detection due to challenges such as object size, complex backgrounds. To address these issues, we propose a robust detection model based on You Only Look Once (YOLO) that balances accuracy and efficiency. The model mainly contains several major innovation: feature selection pyramid network, Inner-Shape Intersection over Union (ISIoU) loss function and small object detection head. To overcome the limitations of traditional fusion methods in handling multi-level features, we introduce a Feature Selection Pyramid Network integrated into the Neck component, which preserves shallow feature details critical for detecting small objects. Additionally, recognizing that deep network structures often neglect or degrade small object features, we design a specialized small object detection head in the shallow layers to enhance detection accuracy for these challenging targets. To effectively model both local and global dependencies, we introduce a Conv-Former module that simulates Transformer mechanisms using a convolutional structure, thereby improving feature enhancement. Furthermore, we employ ISIoU to address object imbalance and scale variation This approach accelerates model conver-gence and improves regression accuracy. Experimental results show that, compared to the baseline model, the proposed method significantly improves small object detection performance on the VisDrone2019 dataset, with mAP@50 increasing by 4.9% and mAP@50-95 rising by 6.7%. This model also outperforms other state-of-the-art algorithms, demonstrating its reliability and effectiveness in both small object detection and remote sensing image fusion tasks.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Video Surveillance and Tracking Methods · Brain Tumor Detection and Classification
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (38)
Show source-provided references
1. Multi-view 3D object detection network for autonomous driving
2. An object classification approach for autonomous vehicles using machine learning techniques10.3390/wevj14020041 ↗
3. Advanced aerial monitoring and vehicle classification for intelligent transportation systems with YOLOv8 variants10.1016/j.jnca.2025.104134 ↗
4. Enhancing vehicle detection in intelligent transportation systems via autonomous UAV platform and YOLOv8 integration10.1016/j.asoc.2024.112015 ↗
5. WildlifeMapper: aerial image analysis for multi-species detection and identification
6. Drone-YOLO: improved YOLO for small object detection in UAV
7. Dense and small object detection in UAV vision based on cascade network
8. Microsoft COCO: common objects in context
9. Infrared small target detection algorithm based on improved dense nested U-Net network10.3390/s25030814 ↗
10. Detection and tracking meet drones challenge10.1109/TPAMI.2021.3119563 ↗
11. Rich feature hierarchies for accurate object detection and semantic segmentation
12. Faster R-CNN: towards real-time object detection with region proposal networks10.1109/TPAMI.2016.2577031 ↗
13. YOLOv8: a novel object detection algorithm with enhanced performance and robustness
14. YOLOv10: real-time end-to-end object detection
15. SSD: single shot multibox detector
16. SOD-YOLOv8-enhancing YOLOv8 for small object detection in aerial imagery and traffic scenes10.3390/s24196209 ↗
17. A small-object detection model based on improved YOLOv8s for UAV image scenarios10.3390/rs16132465 ↗
18. YOLOv8-MDN-tiny: a lightweight model for multi-scale disease detection of postharvest golden passion fruit10.1016/j.postharvbio.2024.113281 ↗
19. Faster and lightweight: an improved YOLOv5 object detector for remote sensing images10.3390/rs15204974 ↗
20. A lightweight fusion strategy with enhanced interlayer feature correlation for small object detection
21. Chen Z, Lu S. CAF-YOLO: a robust framework for multi-scale lesion detection in biomedical imagery. arXiv:2408.01897. 2024.10.1109/ICASSP49660.2025.10888358 ↗
22. A DeNoising FPN with transformer R-CNN for tiny object detection
23. SuperYOLO: super resolution assisted object detection in multimodal remote sensing imagery
24. Small-object detection in remote sensing images with super-resolution perception10.1109/JSTARS.2024.3452707 ↗
25. Zhang T, Kasichainula K, Zhuo Y, Li B, Seo JS, Cao Y. Patch-based selection and refinement for early object detection. arXiv:2311.02274. 2023.10.1109/WACV57701.2024.00078 ↗
26. A multi-feature fusion and attention network for multi-scale object detection in remote sensing images10.3390/rs15082096 ↗
27. Visible and infrared image fusion using deep learning10.1109/TPAMI.2023.3261282 ↗
28. Zhang H, Cisse M, Dauphin YN, Lopez-Paz D. Beyond empirical risk minimization. arXiv:1710. 09412. 2017.10.1007/978-1-4899-7687-1_79 ↗
29. Bowles C, Chen L, Guerrero R, Bentley P, Gunn R, Hammers A, et al. GAN augmentation: augmenting training data using generative adversarial networks. arXiv:1810.10863. 2018.
30. Jackson PT, Atapour-Abarghouei A, Bonner S, Breckon T, Obara B. Style augmentation: data augmentation via style randomization. arXiv:1809.05375. 2018.
31. Kisantal M, Wojna Z, Murawski J, Naruniec J, Cho K. Augmentation for small object detection. arXiv:1902.07296. 2019.10.5121/csit.2019.91713 ↗
32. RRNet: a hybrid detector for object detection in drone-captured images
33. Xu Z, Wu D, Yu C, Chu X, Sang N, Gao C. SCTNet: single-branch CNN with transformer semantic information for real-time segmentation. arXiv:2312.17071. 2023.10.1609/aaai.v38i6.28457 ↗
34. A feature-enhanced small object detection algorithm based on attention mechanism10.3390/s25020589 ↗
35. Zhang H, Xu C, Zhang S. Inner-IoU: more effective intersection over union loss with auxiliary bounding box. 2023. doi:10.48550/arXiv.2311.02877.
36. Ma S, Xu Y, Ma S, Xu Y. MPDIoU: a loss for efficient and accurate bounding box regression. arXiv:2307.07662. 2023.
37. More accurate metric considering bounding box shape and scale
38. A normalized Gaussian Wasserstein distance for tiny object detection