Color Gray Pictures with GAN
environment
pytorch numpy opencv2
Dataset
get dataset, or others。Please separate the pictures of training set and test set manually. And put the pictures of training set to data/train/image, put the test set to data/test/image
How to use
python pre_train.py python train.py python test.py The results will be saved under datas/test/generated
Our result
see loss.txt