A Fast Target Detection Model for Remote Sensing Images Leveraging Roofline Analysis on Edge Computing Devices

被引:0
作者
Zhao, Boya [1 ]
Qin, Zihan [1 ,2 ]
Wu, Yuanfeng [1 ,3 ]
Song, Yuhang [4 ]
Yu, Haoyang [2 ]
Gao, Lianru [1 ]
机构
[1] Chinese Acad Sci, Aerosp Informat Res Inst, Key Lab Computat Opt Imaging Technol, Beijing 100094, Peoples R China
[2] Dalian Maritime Univ, Informat Sci & Technol Coll, Ctr Hyperspectral Imaging Remote Sensing, Dalian 116026, Peoples R China
[3] Chinese Acad Sci, Aerosp Informat Res Inst, Key Lab Digital Earth Sci, Beijing 100094, Peoples R China
[4] Univ Chinese Acad Sci, Coll Resources & Environm, Beijing 100049, Peoples R China
基金
国家重点研发计划;
关键词
Computational modeling; Hardware; Convolution; Object detection; Feature extraction; Remote sensing; Graphics processing units; Rendering (computer graphics); Inference algorithms; Edge computing; remote sensing image; target detection; NETWORK;
D O I
10.1109/JSTARS.2024.3483749
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Deploying image target detection algorithms on embedded devices is critical. Previous studies assumed that fewer model parameters and computations improved the inference speed. However, many models with few parameters and computations have slow inference speeds. Therefore, developing a remote sensing image target detection model that can perform real-time inference on embedded devices is required. We propose a fast target detection model for remote sensing images leveraging roofline analysis on edge computing devices (FTD-RLE). It comprises three parts: (1) We analyze the hardware characteristics of embedded devices using RoofLine and incorporate global features to design a model structure based on the operational intensity (OI) and arithmetic intensity (AI) of embedded devices. (2) The mirror ring convolution (MRC) is designed for extracting global features. The global information-aware module (GIAM) extracts local features from key areas using the global feature guidance model. The global-local feature pyramid module (GLFPM) is proposed to combine global and local features. (3) Additionally, hardware deployment and inference acceleration technologies are implemented to enable the model's deployment on edge devices. TensorRT and quantization methods are used to ensure fast inference speed. The proposed algorithm achieves an average detection accuracy of 92.3% on the VHR-10 dataset and 95.2% on the RSOD dataset. It has 1.26 M model parameters, and the inference time for processing one image on Jetson Orin Nx is 8.43ms, which is 1.90 ms and 1.98 ms faster than the mainstream lightweight algorithms ShufflenetV2 and GhostNet, respectively.
引用
收藏
页码:19343 / 19360
页数:18
相关论文
共 56 条
  • [1] [Anonymous], 2019, REMOTE SENS-BASEL, DOI DOI 10.3390/rs11202376
  • [2] Towards Multi-class Object Detection in Unconstrained Remote Sensing Imagery
    Azimi, Seyed Majid
    Vig, Eleonora
    Bahmanyar, Reza
    Koerner, Marco
    Reinartz, Peter
    [J]. COMPUTER VISION - ACCV 2018, PT III, 2019, 11363 : 150 - 165
  • [3] Real-Time Semantic Segmentation A brief survey and comparative study in remote sensing
    Broni-Bediako, Clifford
    Xia, Junshi
    Yokoya, Naoto
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2023, 11 (04) : 94 - 124
  • [4] Learning Rotation-Invariant Convolutional Neural Networks for Object Detection in VHR Optical Remote Sensing Images
    Cheng, Gong
    Zhou, Peicheng
    Han, Junwei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (12): : 7405 - 7415
  • [5] Histograms of oriented gradients for human detection
    Dalal, N
    Triggs, B
    [J]. 2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, : 886 - 893
  • [6] Model Compression and Hardware Acceleration for Neural Networks: A Comprehensive Survey
    Deng, Lei
    Li, Guoqi
    Han, Song
    Shi, Luping
    Xie, Yuan
    [J]. PROCEEDINGS OF THE IEEE, 2020, 108 (04) : 485 - 532
  • [7] Dettmers T., 2016, P PUERT RIC INT C LE, P1
  • [8] An Instruction Roofline Model for GPUs
    Ding, Nan
    Williams, Samuel
    [J]. PROCEEDINGS OF 2019 IEEE/ACM PERFORMANCE MODELING, BENCHMARKING AND SIMULATION OF HIGH PERFORMANCE COMPUTER SYSTEMS (PMBS 2019), 2019, : 7 - 18
  • [9] Object Detection with Discriminatively Trained Part-Based Models
    Felzenszwalb, Pedro F.
    Girshick, Ross B.
    McAllester, David
    Ramanan, Deva
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2010, 32 (09) : 1627 - 1645
  • [10] Enhanced Feature Representation in Detection for Optical Remote Sensing Images
    Fu, Kun
    Chen, Zhuo
    Zhang, Yue
    Sun, Xian
    [J]. REMOTE SENSING, 2019, 11 (18)