This repository contains the code (in PyTorch) for the paper: (IEEE TGRS)
If you use this code, please cite our paper. Please hit the star at the top-right corner. Thanks!
Spaceborne infrared ship detection holds immense research significance in both military and civilian domains. Nonetheless, the focus of research in this field remains primarily on optical and Synthetic Aperture Radar (SAR) images due to the confidentiality and limited accessibility of infrared data. The challenges in spaceborne ship detection arise from the long-distance capture and low signal-to-noise ratio of infrared images, which contribute to false alarm misclassifications. To handle this problem, this paper concentrates on enhancing information interaction during feature extraction to discern disparities between targets and backgrounds more effectively, and we propose a Multi-Type Feature Perception and Refined Network (MFPRN). Specifically, we propose a dual feature fusion scheme, which combines a Fast Fourier module used to obtain comprehensive receptive field and a lightweight MLP applied to capture the long-range feature dependencies. Besides, we adopt a Cascade Region Proposal Network to leverage high-quality region proposals for the prediction head. Through the extraction of rich features and refined candidate boxes, we successfully mitigate false alarms. Experimental results illustrate that our method significantly reduces false alarms for general detectors, culminating in state-of-the-art performance as demonstrated on the public ISDD baseline.
- Python == 3.7.11
- PyTorch == 1.10.1
- mmdetection == 2.22.0
- mmcv == 1.4.0
- numpy == 1.21.2
The basic installation follows with mmdetection. It is recommended to use manual installation.
ISDD: https://github.com/yaqihan-9898/ISDD
After downloading all datasets, create ISDD document.
$ cd data
$ mkdir ISDD
It is recommended to symlink the dataset root to $data
.
ISDD
├── data
│ ├── ISDD
│ │ ├──VOC2007
│ │ │ ├── JPEGImages
│ │ │ ├── ImgaeSets
│ │ │ ├── Annotations
$ python tools/train.py configs/mfprn/crpn_mfprn_r50_fpn_1x_isdd_voc.py
$ python tools/test.py configs/mfprn/crpn_mfprn_r50_fpn_1x_isdd_voc.py <path/to/checkpoints>
isdd: https://drive.google.com/file/d/1uLUUjuTU1OFsc_6HP9YJeRutRQ2Ssl5w/view?usp=drive_link
Thanks MMDetection team for the wonderful open source project!
@ARTICLE{Yuan2023mfprn,
author={Yuan, Jieyu and Cai, Zhanchuan and Wang, Shiyu and Kong, Xiaoxi},
journal={IEEE Transactions on Geoscience and Remote Sensing},
title={A Multitype Feature Perception and Refined Network for Spaceborne Infrared Ship Detection},
year={2024},
volume={62},
number={},
pages={1-11},
doi={10.1109/TGRS.2023.3341215}}