SCNet: A Lightweight and Efficient Object Detection Network for Remote Sensing

被引:6
作者
Zhu, Shiliang [1 ,2 ]
Miao, Min [3 ]
机构
[1] Beijing Informat Sci & Technol Univ, Key Lab Informat & Commun Syst, Minist Informat Ind, Beijing 100101, Peoples R China
[2] Beijing Informat Sci & Technol Univ, Key Lab Minist Educ Optoelect Measurement Technol, Beijing 100101, Peoples R China
[3] Beijing Informat Sci & Technol Univ, Acad Smart IC & Network, Beijing 100101, Peoples R China
关键词
Lightweight model; object detection; remote sensing; you only look once (YOLO); ATTENTION;
D O I
10.1109/LGRS.2023.3344937
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Detecting small objects in remote sensing images is meaningful challenging, especially when deploying existing object detection models on edge terminal devices with limited hardware resources. In this study, we present an efficient remote sensing object detection model named SCNet, based on the ultra-lightweight (you only look once, YOLOv5n). To address the significant feature loss issue in small objects within the model's neck, we introduce the selective feature enhancement block (SFEB). The SFEB selectively processes a portion of feature maps that contribute more to semantic information extraction while retaining another portion, enabling us to extract rich semantic information while preserving crucial details information necessary for small object detection. Furthermore, we incorporate the contextual transformer block (CTB) at the neck and backbone junction, which enhances the model's ability to understand relationships and boundaries between objects and backgrounds by exploring contextual information in shallow-level feature maps. This improves the model's capability to detect challenging small and medium objects. Experimental results on the NWPU VHR-10 and DIOR datasets demonstrate the model's performance, achieving mean average precisions (mAPs) of 96.6% and 72.6% at IOU = 0.5. The model operates at 487 frames/s with a batch size of 32 (FPS32), requiring only 4.6 giga floating-point operations per second (GFLOPs) and 1.8 million params.
引用
收藏
页码:1 / 5
页数:5
相关论文
共 20 条
  • [1] Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks
    Chen, Jierun
    Kao, Shiu-Hong
    He, Hao
    Zhuo, Weipeng
    Wen, Song
    Lee, Chul-Ho
    Chan, S. -H. Gary
    [J]. 2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 12021 - 12031
  • [2] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [3] Jia Liu, 2020, 2020 IEEE 22nd International Conference on High Performance Computing and Communications
  • [4] IEEE 18th International Conference on Smart City
  • [5] IEEE 6th International Conference on Data Science and Systems (HPCC/SmartCity/DSS), P585, DOI 10.1109/HPCC-SmartCity-DSS50907.2020.00074
  • [6] Contextual Transformer Networks for Visual Recognition
    Li, Yehao
    Yao, Ting
    Pan, Yingwei
    Mei, Tao
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (02) : 1489 - 1500
  • [7] YOLO-DA: An Efficient YOLO-Based Detector for Remote Sensing Object Detection
    Lin, Jiehua
    Zhao, Yan
    Wang, Shigang
    Tang, Yu
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [8] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37
  • [9] Mingxing Tan, 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). Proceedings, P10778, DOI 10.1109/CVPR42600.2020.01079
  • [10] Lightweight Oriented Object Detection Using Multiscale Context and Enhanced Channel Attention in Remote Sensing Images
    Ran, Qiong
    Wang, Qing
    Zhao, Boya
    Wu, Yuanfeng
    Pu, Shengliang
    Li, Zijin
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 (14) : 5786 - 5795