计算机科学
目标检测
计算机视觉
对象(语法)
遥感
人工智能
模式识别(心理学)
地质学
标识
DOI:10.1109/mvipit65697.2024.00018
摘要
This paper introduces RS-YOLO, a novel model for small object detection in remote sensing, designed to address the challenges of missed and false detections in scenarios with densely distributed, small objects. By integrating the Biformer, a dynamic sparse attention mechanism with low computational cost, the model can better suppress background noise and enhance the ability to detect small objects. We also utilize an efficient neck structure, Gold-YOLO, based on the Gather-and-Distribute (GD) mechanism to enhance feature fusion capabilities. Furthermore, we combine Inner-IoU and CloU loss functions as a new bounding box regression (BBR) loss function, enhancing localization performance and model robustness while accelerating training. On AI-TOD and DIOR datasets, RS-YOLO achieves mAPO.5 scores of 56.3 % and 74.2 %, outperforming baseline models and other mainstream algorithms.
科研通智能强力驱动
Strongly Powered by AbleSci AI