Environment: Tested with PyTorch 1.8.1 and CUDA 10.2.
-
Create and activate a conda environment:
conda create -n rdg source activate rdg -
Install dependencies:
pip install torch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 yacs openpyxl matplotlib pandas segmentation_models_pytorch albumentations
Dataset:
- Potsdam and Bhopal datasets are used for training.
- Download the dataset from here.
Training:
sh bashes/train_rdg.sh- Acknowledgement
- The code is based on ResiDualGAN-DRDG.