JasonWZC / BASNet

Code for BASNet

BASNet

Environment setup

cuda: 11.7  
python: 3.9  
pytorch: 1.13.1  

Installation

git clone https://github.com/JasonWZC/BASNet-main.git 
cd BASNet-main 

Train

You can set the training model and parameters in BASNet/train.py.

python train.py

Prediction

After training, you can put weight in BASNet/predict/plot_demo.py.
Then use it to generate a prediction graph.

python plot_demo.py

Download the datesets

• LEVIR-CD: LEVIR-CD

• CDD: CDD

• GZ-CD: GZ-CD

You can install these datasets to './datasets'. And we have provided a small dataset called TEST for your testing.