search

pfnet / PaintsChainer

line drawing colorization using chainer
https://paintschainer.preferred.tech/
MIT License
3.77k stars 557 forks source link

What should I set for the leak variable in the constructor of Image2ImageDataset? #99

Closed mike199212 closed 7 years ago

mike199212 commented 7 years ago

In the training of cnn_128, the default value for "leak" of Imae2ImageDataset is set to (0,0). However, one needs to set leak[1] to some positive value to let cnn_128 to be able to respond to "ref" (hint colors). What value should I use, or how can I find appropriate value?

taizan commented 7 years ago

Yes, and it is depends on density of target ref image. I m also not sure about appropriate value, but I used 100 ~ 500 for leak[1]

mike199212 commented 7 years ago

Thank you for your reply! Could you tell me about your leak[0] and leak[1] to train the second network cnn_x2, i.e. the tuple "leak" in the constructor of Image2ImageDatasetX2?

taizan commented 7 years ago

in my opinion, 8 ~ 16 and 64~128 is ok for x2 leak[0] and leak[1] . but 2nd layer is still not highly optimized.

mike199212 commented 7 years ago

Thank you again. I will try using these values in training second network.