Multispectral conditional Generative Adversarial Nets

This repository is an implementation of "Filmy Cloud Removal on Satellite Imagery with Multispectral Conditional Generative Adversarial Nets".

Requirements

I recommend Anaconda to manage your Python libraries.
Because it is easy to install some of the libraries necessary to prepare the data.

Python3 (tested with 3.5.4)
PyTorch (tested with 0.4.1)
TorchVision (tested with 0.2.1)
Numpy (tested with 1.14.2)
OpenCV (tested with 3.3.1)
Pillow (tested with 5.0.0)
tqdm (tested with 4.15.0)
PyYAML (tested with 3.12)

Preparing the data

Please refer to make_dataset/README.md.

How to train

You need set each parameters in config.yml.
When you run train.py, config.yml is automatically copied to a directory out_dir defined at config.yml.

python train.py

How to test

python predict.py --config <path_to_config.yml_in_the_out_dir> --test_dir <path_to_a_directory_stored_test_data> --out_dir <path_to_an_output_directory> --pretrained <path_to_a_pretrained_model> --cuda

Pre-trained model

You can download a pre-trained model from here. (200MB)

License

Academic use only.