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 |