our paper " Image Recoloring for Color Vision Deficiency Compensation Using Swin Transformer". CVD, recoloring
conda env create -f 190pytorch_environment.yml
https://drive.google.com/drive/folders/10WMXPbtpV7Hy5_qBA_TCEbW-kCpj1D7v?usp=sharing
put the dataset in the same directory.
python train.py --epoch 0 --n_epochs 121 --batch_size 8 --checkpoint_interval 10 --dataset_name cvd_100_001 --sample 5000 --ssimori 1 --cvd 1 --lambda_ssim 0.00 --points 3000
the model will be saved in ./saved_models the images will be saved in ./images