改进YOLO检测小目标,精度提升超25点。
Enhancing Small Object Detection with YOLO: A Novel Framework for Improved Accuracy and Efficiency
- 优化滑动窗口裁剪参数并增强网络结构
- 在VisDrone2019上将mAP50-50从35.5提升至61.2
- 适合遥感图像中微小目标检测任务
本文针对大规模航拍图像中小目标检测问题,提出一种新型框架。现有方法多依赖图像裁剪与网络架构改进,如滑动窗口裁剪、高分辨率特征图和注意力机制。为提升检测效率与精度,研究基于SW-YOLO,优化滑动窗口的裁剪尺寸与重叠率,并通过结构改进:在颈部引入先进特征提取模块以增强特征图,在骨干网络中集成CBAM保留空间与通道信息,并设计新检测头以提升小目标识别能力。与SAHI及CZDet等先进方法对比,所提模型在VisDrone2019数据集上表现优异,相较基线YOLOv5L的mAP50-50从35.5显著提升至61.2,优于CZDet的58.36,验证了该框架的有效性与鲁棒性。
原文摘要 · Abstract (English)
This paper investigates and develops methods for detecting small objects in large-scale aerial images. Current approaches for detecting small objects in aerial images often involve image cropping and modifications to detector network architectures. Techniques such as sliding window cropping and architectural enhancements, including higher-resolution feature maps and attention mechanisms, are commonly employed. Given the growing importance of aerial imagery in various critical and industrial applications, the need for robust frameworks for small object detection becomes imperative. To address this need, we adopted the base SW-YOLO approach to enhance speed and accuracy in small object detection by refining cropping dimensions and overlap in sliding window usage and subsequently enhanced it through architectural modifications. we propose a novel model by modifying the base model architecture, including advanced feature extraction modules in the neck for feature map enhancement, integrating CBAM in the backbone to preserve spatial and channel information, and introducing a new head to boost small object detection accuracy. Finally, we compared our method with SAHI, one of the most powerful frameworks for processing large-scale images, and CZDet, which is also based on image cropping, achieving significant improvements in accuracy. The proposed model achieves significant accuracy gains on the VisDrone2019 dataset, outperforming baseline YOLOv5L detection by a substantial margin. Specifically, the final proposed model elevates the mAP .5.5 accuracy on the VisDrone2019 dataset from the base accuracy of 35.5 achieved by the YOLOv5L detector to 61.2. Notably, the accuracy of CZDet, which is another classic method applied to this dataset, is 58.36. This research demonstrates a significant improvement, achieving an increase in accuracy from 35.5 to 61.2.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。