Painting is sketch-colorization project leveraging various GANs.
Note that I've not got good result yet.
This project is powered by following packages.
hydra is prerelease version which is installed by below command.
$ pip install hydra-core --upgrade --pre
-
Baseline
Normal colorization model. -
Lightweight
Using LightweightGAN's training procedure.
Reference: open-review
Before run training script, you have to make "configs/env/defaults_env.yaml" like below.
data_dir: path/to/<image-dir>
save_dir: path/to/<save-dir>
Run training script.
$ python train.py method=<*method-name*>
<*method-name*>: e.g. baseline, which corresponds to file name of "confgs/method/*.yaml"