watercolor-CycleGAN(not stable)

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

For the good watercolor drawings.

Data

Landscape: I used kaggle competition data.

Watercolor: I used WikiArt

Requirements

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

Checkpoint

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

Latest generated images(Original - 350 epochs - 550 epochs)

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