Remote Sensing Image Change Detection with Graph Interaction

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

Graph Interaction Module (GIM)

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/BGINet.git
cd BGINet

Quick Start

We have some samples from the WHU dataset in the folder samples for a quick start.

Firstly, you can download our BGINet pretrained model

WHU-CD： baidu drive, code: afse .

GZ-CD： baidu drive, code: 3knv .

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

The WHU Building Change Detection Dataset :The dataconsists of two aerial images of two different time phases and the exact location, which contains $12796$ buildings in $20.5km^2$ with a resolution of $0.2 m$ and a size of $32570\times15354$.We crop the images to $256\times256$ size and randomly divide the training, validation, and test sets:$ 6096/762/762$. Guangzhou Dataset(GZ-CD) : The dataset was collectedfrom $2006-2019$, covering the suburbs of Guangzhou, China, and to facilitate the generation of image pairs, the Google Earth service of BIGEMAP software was used to collect 19 seasonally varying VHR image pairs with a spatial resolu- tion of$ 0.55 m$ and a size range of $1006\times1168$ pixels to $4936\times5224$.We crop the images to $256\times 256$ size and randomly divide the training, validation, and test sets:$ 2876/353/374$

Dataset	Name	Link
GZ-CD building change detection dataset	`GZ`	website
WHU building change detection dataset	`WHU`	website

License

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