Skip to content

A general framework for change detection of remote sensing images

Notifications You must be signed in to change notification settings

wangle53/CSA-CDGAN

Repository files navigation

A general framework for change detection of remote sensing images
paper link: https://link.springer.com/article/10.1007/s00521-022-07637-z

Quantitative & Qualitative Results on CDD, WHU-CD, LEVIR-CD Datasets

Requirements

Python 3.7.0  
Pytorch 1.6.0  
Visdom 0.1.8.9  
Torchvision 0.7.0

Datasets

Pretrained Model

Pretrained models for CDD, LEVIR-CD and WHU-CD are available. You can download them from the following link. [Baiduyun] the password is yudl. [GoogleDrive]

Test

Before test, please download datasets and pretrained models. Revise the data-path in constants.py to your path. Copy pretrained models to folder './dataset_name/outputs/best_weights', and run the following command:

cd CSA-CDGAN_ROOT
python make_dataset.py
python test.py

make_dataset.py can generate .txt files for training, validation and test. Not that the dataset structure should be the same as following:

Custom dataset
|--train
  |--file1
    |--t0.jpg, t1.jpt, label.jpg
  |--file2
    |--t0.jpg, t1.jpt, label.jpg
  ...
|--test
  |--file1
    |--t0.jpg, t1.jpt, label.jpg
  |--file2
    |--t0.jpg, t1.jpt, label.jpg
  ...
|--validation
  |--file1
    |--t0.jpg, t1.jpt, label.jpg
  |--file2
    |--t0.jpg, t1.jpt, label.jpg
  ...

Training

cd CSA-CDGAN_ROOT
python make_dataset.py
python -m visdom.server
python train.py

To display training processing, copy 'http://localhost:8097' to your browser.

Citing CSA-CDGAN

If you use this repository or would like to refer the paper, please use the following BibTex entry.

@article{wang2022csa,
  title={CSA-CDGAN: channel self-attention-based generative adversarial network for change detection of remote sensing images},
  author={Wang, Zhixue and Zhang, Yu and Luo, Lin and Wang, Nan},
  journal={Neural Computing and Applications},
  pages={1--15},
  year={2022},
  publisher={Springer}
}

Reference

-Akcay, Samet, Amir Atapour-Abarghouei, and Toby P. Breckon. "Ganomaly: Semi-supervised anomaly detection via adversarial training." Asian conference on computer vision. Springer, Cham, 2018.

More

My personal google web

About

A general framework for change detection of remote sensing images

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages