watercolor-CycleGAN(not stable)

This is my personal project, if you are insterested in or just wanna ask some questions dont be afraid.

Data

pip3 install torch
pip3 install pytorch_lightning
pip3 install torchvision
pip3 install numpy
pip3 install pandas
pip3 install matplotlib

I couldn't upload the checkpoint to github cause the size is so big and my internet is to poor.

Google Drive link for 550 epochs

Probably need more epoch or more data, maybe higher learning rate.

Name		Name	Last commit message	Last commit date
Latest commit History 35 Commits
DATA		DATA
readme		readme
CycleGAN_ls.py		CycleGAN_ls.py
LICENSE		LICENSE
README.md		README.md
dataModule.py		dataModule.py
discriminator.py		discriminator.py
generator.py		generator.py
imageGenerate.py		imageGenerate.py
model_utils.py		model_utils.py
train.py		train.py