Home

Awesome

BSINet-CD: Bitemporal Semantics Interaction Network for Remote Sensing Images Change Detection

Here, we provide the pytorch implementation of the paper: Remote Sensing Image Change Detection with Graph Interaction

For more ore information, please see our published paper at arxiv.

Overall Architecture

image-20230827110536151

Semantics Interaction Module (SIM)

image-20230706103041124

Requirements

albumentations>=1.3.0
numpy>=1.20.2
opencv_python>=4.7.0.72
opencv_python_headless>=4.7.0.72
Pillow>=9.4.0
Pillow>=9.5.0
scikit_learn>=1.0.2
torch>=1.9.0
torchvision>=0.10.0

Installation

Clone this repo:

git clone https://github.com/JackLiu-97/BSINet.git
cd BSINet

Quick Start

Firstly, you can download our BSINet pretrained model

WHU-CD: baidu drive, code: afse .

GZ-CD: baidu drive, code: afse .

After downloaded the pretrained model, you can put it in output.

Then, run a demo to get started as follows:

python demo.py --ckpt_url ${model_path} --data_path ${sample_data_path} --out_path ${save_path}

After that, you can find the prediction results in ${save_path}.

Train

To train a model from scratch, use

python train.py --data_path ${train_data_path} --val_path ${val_data_path} --lr ${lr} --batch_size ${-batch_size} 

Evaluate

To evaluate a model on the test subset, use

python test.py --ckpt_url ${model_path} --data_path ${test_data_path}

Supported Datasets

DatasetNameLink
GZ-CD building change detection datasetGZwebsite
WHU building change detection datasetWHUwebsite

License

Code is released for non-commercial and research purposes only. For commercial purposes, please contact the authors.