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.
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
Clone this repo:
git clone https://github.com/JackLiu-97/BSINet.git
cd BSINet
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}
.
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}
To evaluate a model on the test subset, use
python test.py --ckpt_url ${model_path} --data_path ${test_data_path}
-
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 resolution 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 |
Code is released for non-commercial and research purposes only. For commercial purposes, please contact the authors.