/LR-TSDet

LR-TSDet: Towards Tiny Ship Detection in Low-Resolution Remote Sensing Images (Remote Sens. 2021)

Primary LanguagePythonApache License 2.0Apache-2.0

LR-TSDet: Towards Tiny Ship Detection in Low-Resolution Remote Sensing Images

Official implementation of LR-TSDet. This repository is based on mmdetection.

LR-TSDet: Towards Tiny Ship Detection in Low-Resolution Remote Sensing Images
Remote Sens. 2021.
Jixiang Wu, Zongxu Pan, Bin Lei, Yuxin Hu.

Abstract

Recently, deep learning-based methods have made great improvements in object detection in remote sensing images (RSIs). However, detecting tiny objects in low-resolution images is still challenging. The features of these objects are not distinguishable enough due to their tiny size and confusing backgrounds and can be easily lost as the network deepens or downsamples. To address these issues, we propose an effective Tiny Ship Detector for Low-Resolution RSIs, abbreviated as LR-TSDet, consisting of three key components: a filtered feature aggregation (FFA) module, a hierarchical-atrous spatial pyramid (HASP) module, and an IoU-Joint loss. The FFA module captures long-range dependencies by calculating the similarity matrix so as to strengthen the responses of instances. The HASP module obtains deep semantic information while maintaining the resolution of feature maps by aggregating four parallel hierarchical-atrous convolution blocks of different dilation rates. The IoU-Joint loss is proposed to alleviate the inconsistency between classification and regression tasks, and guides the network to focus on samples that have both high localization accuracy and high confidence. Furthermore, we introduce a new dataset called GF1-LRSD collected from the Gaofen–1 satellite for tiny ship detection in low-resolution RSIs. The resolution of images is 16m and the mean size of objects is about 10.9 pixels, which are much smaller than public RSI datasets. Extensive experiments on GF1-LRSD and DOTA-Ship show that our method outperforms several competitors, proving its effectiveness and generality.

framework

Updates

  • 2021-09. Code is released.

Usage

Please refer to mmdetection for installation, training, evaluation, etc.

Citation

If you find this repository/work helpful in your research, welcome to cite the paper.

@article{wu2021lrtsdet,
  AUTHOR = {Wu, Jixiang and Pan, Zongxu and Lei, Bin and Hu, Yuxin},
  TITLE = {LR-TSDet: Towards Tiny Ship Detection in Low-Resolution Remote Sensing Images},
  JOURNAL = {Remote Sensing},
  VOLUME = {13},
  YEAR = {2021},
  NUMBER = {19},
  ARTICLE-NUMBER = {3890},
  URL = {https://www.mdpi.com/2072-4292/13/19/3890},
  ISSN = {2072-4292},
}