Awesome
AttentionedDeepPaint
Automatic Anime Sketch Colorization Using Generative Adversarial Network (GAN)
Dependency
Refer requirements
Install
- Pytorch (>= 0.4.1) (No problem in torch 1.0.1)
- Torchvision (>= 0.2.1) based on your python version, os, cuda version etc...
Usage
Download dataset
- go to link and download
- unzip
Train
$ ./train.sh
Reference
- Sketch Image Generation
- Color Histogram Extraction
- Attentioned UNet
Colorize Result Examples
To see experiment trials, click
General Result Example (Same Sketch, Different Styles)
Attention Map Result Examples
Train Settings & Log
Hyperparameter | Value |
---|---|
learning_rate | 0.0002 |
beta1 (optimizer) | 0.5 |
batch_size | 4 |
lambda | 100 |
epochs | 30 |
learning_rate schedule | None |
Discriminator | PatchGAN |
Weight Initialization | (0, 0.02) Normal Distribution |