SMN-YOLO: Lightweight YOLOv8-Based Model for Small Object Detection in Remote Sensing Images

被引:2
作者
Zheng, Xiangyue [1 ,2 ,3 ]
Bi, Jingxin [1 ,2 ,3 ]
Li, Keda [1 ,2 ,3 ]
Zhang, Gang [1 ,2 ,3 ]
Jiang, Ping [1 ,2 ,3 ]
机构
[1] Chinese Acad Sci, Inst Opt & Elect, Chengdu 610209, Peoples R China
[2] Natl Lab Adapt Opt, Chengdu 610209, Peoples R China
[3] Univ Chinese Acad Sci, Beijing 101408, Peoples R China
关键词
Object detection; Remote sensing; Feature extraction; Computational modeling; Accuracy; Training; Attention mechanisms; Standards; Spatial resolution; Semantics; Multiscale feature attention module (MSFAM); remote sensing; small object detection; spatial-channel decoupled downsampling (SCDown);
D O I
10.1109/LGRS.2025.3546034
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
The remote sensing image object detection has advanced significantly; yet, small object detection remains challenging due to their limited size and varying scales. Furthermore, real-world deployment often requires algorithms optimized for fewer parameters and faster inference. To address these issues, we propose SMN-YOLO, a lightweight small object detector based on YOLOv8n. Our approach introduces spatial-channel decoupling downsampling to reduce model size while retaining crucial downsampling information. We also present lightweight and efficient feature pyramid network (LEFPN), a lightweight multiscale feature fusion network incorporating coordinate attention (CA) to capture spatial location cues, enhancing small object detection. In addition, a multiscale feature attention module (MSFAM) further strengthens feature representation. To improve accuracy, we integrate new complete intersection over union (N-CIoU) bounding box regression loss, which minimizes the impact of positional changes on IoU, helping the model focus on low-IoU objects. Experimental results on the vehicle detection in aerial imagery (VEDAI) and AI-based tiny object detection (AI-TOD) datasets show that SMN-YOLO outperforms baseline models with a 3.2% and 2.9% improvement in mean average precision (mAP) at 0.5, respectively, while significantly reducing parameters and only slightly increasing inference time. The proposed model achieves a strong balance between performance and complexity, surpassing several leading detection models.
引用
收藏
页数:5
相关论文
共 22 条
[1]   Towards Large-Scale Small Object Detection: Survey and Benchmarks [J].
Cheng, Gong ;
Yuan, Xiang ;
Yao, Xiwen ;
Yan, Kebing ;
Zeng, Qinghua ;
Xie, Xingxing ;
Han, Junwei .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (11) :13467-13488
[2]   Coordinate Attention for Efficient Mobile Network Design [J].
Hou, Qibin ;
Zhou, Daquan ;
Feng, Jiashi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13708-13717
[3]  
Jocher G., 2024, Ultralytics YOLO
[4]   Drone-DETR: Efficient Small Object Detection for Remote Sensing Image Using Enhanced RT-DETR Model [J].
Kong, Yaning ;
Shang, Xiangfeng ;
Jia, Shijie .
SENSORS, 2024, 24 (17)
[5]  
Li CY, 2022, Arxiv, DOI arXiv:2209.02976
[6]   An Improved S2A-Net Algorithm for Ship Object Detection in Optical Remote Sensing Images [J].
Li, Jianfeng ;
Chen, Mingxu ;
Hou, Siyuan ;
Wang, Yongling ;
Luo, Qinghua ;
Wang, Chenxu .
REMOTE SENSING, 2023, 15 (18)
[7]   GDRS-YOLO: More Efficient Multiscale Features Fusion Object Detector for Remote Sensing Images [J].
Lin, Yulong ;
Li, Jun ;
Shen, Shulong ;
Wang, Hong ;
Zhou, Hongzhi .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 :1-5
[8]  
Lv W., 2024, arXiv, DOI [10.48550/arXiv.2407.17140, DOI 10.48550/ARXIV.2407.17140]
[9]  
Pham Vung, 2024, 2024 IEEE International Conference on Big Data (BigData), P8460, DOI 10.1109/BigData62323.2024.10825509
[10]   Vehicle detection in aerial imagery : A small target detection benchmark [J].
Razakarivony, Sebastien ;
Jurie, Frederic .
JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2016, 34 :187-203