English

Simplifying Two-Stage Detectors for On-Device Inference in Remote Sensing

Computer Vision and Pattern Recognition 2024-04-12 v1

Abstract

Deep learning has been successfully applied to object detection from remotely sensed images. Images are typically processed on the ground rather than on-board due to the computation power of the ground system. Such offloaded processing causes delays in acquiring target mission information, which hinders its application to real-time use cases. For on-device object detection, researches have been conducted on designing efficient detectors or model compression to reduce inference latency. However, highly accurate two-stage detectors still need further exploitation for acceleration. In this paper, we propose a model simplification method for two-stage object detectors. Instead of constructing a general feature pyramid, we utilize only one feature extraction in the two-stage detector. To compensate for the accuracy drop, we apply a high pass filter to the RPN's score map. Our approach is applicable to any two-stage detector using a feature pyramid network. In the experiments with state-of-the-art two-stage detectors such as ReDet, Oriented-RCNN, and LSKNet, our method reduced computation costs upto 61.2% with the accuracy loss within 2.1% on the DOTAv1.5 dataset. Source code will be released.

Keywords

Cite

@article{arxiv.2404.07405,
  title  = {Simplifying Two-Stage Detectors for On-Device Inference in Remote Sensing},
  author = {Jaemin Kang and Hoeseok Yang and Hyungshin Kim},
  journal= {arXiv preprint arXiv:2404.07405},
  year   = {2024}
}
R2 v1 2026-06-28T15:50:36.235Z