README.md

November 27, 2025 · View on GitHub

CDMamba: Incorporating Local Clues into Mamba for Remote Sensing Image Binary Change Detection

TGRS

Introduction

Notice🐍🐍: CDMamba has been accepted by IEEE TGRS!

The current branch has been tested on Linux system, PyTorch 2.1.0 and CUDA 12.1, supports Python 3.10.

If you find this project helpful, please give us a ⭐️star⭐️, your support is our greatest motivation.

Updates

🌟 2024.06.20 Released the CDMamba project.

TODO

Updated more change detection methods

Introduction
Benchmark
TODO
Table of Contents
Installation
Dataset Preparation
Model Training and Testing
Citation
License
Contact Us

Installation

Requirements

Linux system, Windows is not tested, depending on whether causal-conv1d and mamba-ssm can be installed
Python 3.8+, recommended 3.10
PyTorch 2.0 or higher, recommended 2.1.0
CUDA 11.7 or higher, recommended 12.1

It is recommended to use Miniconda for installation. The following commands will create a virtual environment named cd_mamba and install PyTorch. In the following installation steps, the default installed CUDA version is 12.1. If your CUDA version is not 12.1, please modify it according to the actual situation.

Note: If you are experienced with PyTorch and have already installed it, you can skip to the next section. Otherwise, you can follow the steps below.

Step 0: Install Miniconda.

Step 1: Create a virtual environment named cd_mamba and activate it.

conda create -n cd_mamba python=3.10
conda activate cd_mamba

Step 2: Install dependencies.

pip install -r requirements.txt

Note: Please refer to https://github.com/hustvl/Vim or https://blog.csdn.net/weixin_45667052/article/details/136311600 when installing mamba.

Install CDMamba

You can download or clone the CDMamba repository.

git clone git@github.com:zmoka-zht/CDMamba.git
cd CDMamba

Dataset Preparation

Remote Sensing Change Detection Dataset

We provide the method of preparing the remote sensing change detection dataset used in the paper.

WHU-CD Dataset

Data download link: WHU-CD Dataset PanBaiDu. Code:t2sb

LEVIR-CD Dataset

Data download link: LEVIR-CD Dataset PanBaiDu. Code:qlvs

LEVIR+-CD Dataset

Data download link: LEVIR+-CD Dataset PanBaiDu. Code: xtj8

Organization Method

You can also choose other sources to download the data, but you need to organize the dataset in the following format：

${DATASET_ROOT} # Dataset root directory, for example: /home/username/data/LEVIR-CD
├── A
│   ├── train_1_1.png
│   ├── train_1_2.png
│   ├──...
│   ├── val_1_1.png
│   ├── val_1_2.png
│   ├──...
│   ├── test_1_1.png
│   ├── test_1_2.png
│   └── ...
├── B
│   ├── train_1_1.png
│   ├── train_1_2.png
│   ├──...
│   ├── val_1_1.png
│   ├── val_1_2.png
│   ├──...
│   ├── test_1_1.png
│   ├── test_1_2.png
│   └── ...
├── label
│   ├── train_1_1.png
│   ├── train_1_2.png
│   ├──...
│   ├── val_1_1.png
│   ├── val_1_2.png
│   ├──...
│   ├── test_1_1.png
│   ├── test_1_2.png
│   └── ...
├── list
│   ├── train.txt
│   ├── val.txt
│   └── test.txt

Model Training and Testing

All configuration for model training and testing are stored in the local folder config

Example of Training on LEVIR-CD Dataset

python train.py --config/mamba/levir_cdmamba.json

Example of Testing on LEVIR-CD Dataset

python test.py --config/mamba/levir_test_cdmamba.json

CDMamba Weight

PanBaiDu download link: Weight PanBaiDu. Code:ckpt

Google Drive download link [https://drive.google.com/file/d/1ImTvjN-vPnlJtVwfemzeHWcjoNMFsrS7/view?usp=drive_link]

Citation

If you use the code or performance benchmarks of this project in your research, please refer to the following bibtex citation of CDMamba.

@ARTICLE{10902569,
  author={Zhang, Haotian and Chen, Keyan and Liu, Chenyang and Chen, Hao and Zou, Zhengxia and Shi, Zhenwei},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={CDMamba: Incorporating Local Clues Into Mamba for Remote Sensing Image Binary Change Detection}, 
  year={2025},
  volume={63},
  number={},
  pages={1-16},
  keywords={Feature extraction;Transformers;Remote sensing;Convolutional neural networks;Visualization;Artificial intelligence;Spatiotemporal phenomena;Computational modeling;Attention mechanisms;Computer vision;Bi-temporal interaction;change detection (CD);high-resolution optical remote sensing image;Mamba;state-space model},
  doi={10.1109/TGRS.2025.3545012}}

@ARTICLE{11268372,
  author={Zhang, Haotian and Guo, Han and Chen, Keyan and Chen, Hao and Zou, Zhengxia and Shi, Zhenwei},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={FoBa: A Foreground-Background co-Guided Method and New Benchmark for Remote Sensing Semantic Change Detection}, 
  year={2025},
  volume={},
  number={},
  pages={1-1},
  keywords={Semantics;Remote sensing;Transformers;Feature extraction;Annotations;Roads;Multitasking;Spatial resolution;Landsat;Land surface;Semantic change detection (SCD);foreground-background co-guided;bi-temporal interaction;mamba;new benchmark},
  doi={10.1109/TGRS.2025.3636947}}

@ARTICLE{10471555,
  author={Zhang, Haotian and Chen, Hao and Zhou, Chenyao and Chen, Keyan and Liu, Chenyang and Zou, Zhengxia and Shi, Zhenwei},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={BiFA: Remote Sensing Image Change Detection With Bitemporal Feature Alignment}, 
  year={2024},
  volume={62},
  number={},
  pages={1-17},
  keywords={Feature extraction;Task analysis;Remote sensing;Transformers;Interference;Decoding;Optical flow;Bitemporal interaction (BI);change detection (CD);feature alignment;flow field;high-resolution optical remote sensing image;implicit neural representation},
  doi={10.1109/TGRS.2024.3376673}}

License

This project is licensed under the Apache 2.0 License.

Contact Us

If you have any other questions❓, please contact us in time 👬