Skip to content

Zhao-Tian-yi/RSDet

Repository files navigation

(TITS 2026) RSDet

Removal Then Selection: A Coarse-to-Fine Fusion Perspective for RGB-Infrared Object Detection

image-20240312011746031

Citation

Paper Link: IEEE Trans-ITS

@ARTICLE{11278552,
  author={Zhao, Tianyi and Yuan, Maoxun and Jiang, Feng and Wang, Nan and Wei, Xingxing},
  journal={IEEE Transactions on Intelligent Transportation Systems}, 
  title={Removal Then Selection: A Coarse-to-Fine Fusion Perspective for RGB-Infrared Object Detection}, 
  year={2026},
  volume={27},
  number={2},
  pages={2504-2519},
  keywords={Object detection;Feature extraction;Lighting;Detectors;Information filters;Filtering theory;Representation learning;Attenuation;Reliability;Location awareness;Coarse-to-fine fusion;mixture of experts;multisensory fusion;RGB-IR object detection},
  doi={10.1109/TITS.2025.3638627}}

Getting Started

Installation

Refer to the official mmdet documentation : mmdetection installation

Step 1: Clone the RSDet repository:

To get started, first clone the RSDet repository and navigate to the project directory:

git clone https://github.com/Zhao-Tian-yi/RSDet.git
cd RSDet

Step 2: Environment Setup:

RSDet recommends setting up a conda environment and installing dependencies via pip. Use the following commands to set up your environment:

Create and activate a new conda environment

conda create -n RSDet python=3.7
conda activate RSDet

If you develop and run mmdet directly, install it from source

pip install -v -e .

Install Dependencies

pip install -r requirements.txt
pip install -r requirements_rgbt.txt

Data Preparation

This repository uses relative dataset paths by default. Place datasets under data/ or update the corresponding data_root in the dataset config files under configs/_base_/datasets.

Suggested directory layout:

RSDet/
├── Datasets_Dir/
│   ├── FLIR_align/
│   ├── LLVIP/
│   ├── M3FD/
│   ├── MFAD/
│   └── KAIST/
└── pretrain/
    └── resnet50_cityscape.pth

dataset base configs:

Pretrained Backbone

The released RSDet configs expect the backbone checkpoint at: Download Link

pretrain/resnet50_cityscape.pth

If you want to use a different pretrained checkpoint, update the pretrained_backbone variable in the corresponding config file.

Available RSDet Configs

This branch currently includes the following paper-aligned RSDet configs:

Training

./tools/dist_train.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_FLIR.py 4
./tools/dist_train.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_KAIST.py 4
./tools/dist_train.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_LLVIP.py 4
./tools/dist_train.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_M3FD.py 4
./tools/dist_train.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_MFAD.py 4

Testing

You can download our trained model weights for testing: Model CKPT Download Links.

./tools/dist_test.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_FLIR.py /path/to/checkpoint.pth 4
./tools/dist_test.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_LLVIP.py /path/to/checkpoint.pth 4
./tools/dist_test.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_M3FD.py /path/to/checkpoint.pth 4
./tools/dist_test.sh configs/fusion/RSDet/faster_rcnn_r50_rsdet_MFAD.py /path/to/checkpoint.pth 4

KAIST detection results of RSDet method: txt files, open the KAISTdevkit-matlab-wrapper and run the demo_test.m.

Dataset Layout

The repository assumes that each dataset contains paired visible and infrared images, together with COCO-style annotation files.

Recommended directory layout:

RSDet/
├── Datasets_Dir/
│   ├── FLIR_align/
│   │   ├── train/
│   │   ├── test/
│   │   ├── Annotation_train.json
│   │   └── Annotation_test.json
│   ├── LLVIP/
│   │   ├── train/
│   │   ├── test/
│   │   ├── Annotation_train.json
│   │   └── Annotation_test.json
│   ├── M3FD/
│   │   ├── train/
│   │   ├── test/
│   │   ├── Annotation_train.json
│   │   └── Annotation_test.json
│   ├── MFAD/
│   │   ├── train/
│   │   ├── test/
│   │   ├── Annotation_train.json
│   │   └── Annotation_test.json
│   └── KAIST/
│       ├── kaist_train/
│       ├── kaist_test/
│       ├── kaist_train_data.json
│       └── kaist_test_data.json
└── pretrain/
    └── resnet50_cityscape.pth

Paths To Modify

If your local directory layout is different, update the following paths before training:

Acknowledgment

RSDet is built upon the excellent open-source framework:

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors