Skip to content

Rmote Sensing Image Change Detection With Graph Interaction

Notifications You must be signed in to change notification settings

JackLiu-97/BSINet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

42 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

  • 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

License

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

About

Rmote Sensing Image Change Detection With Graph Interaction

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages