Output is grayscale about coloring-greyscale-images HOT 3 CLOSED

I tried too, the output is also grayscale
https://github.com/microic/niy/tree/master/examples/colorizing_photos

from coloring-greyscale-images.

It's tricky to get good results. A lot of has to do with how many epochs you train it and which training data you use. I'd recommend starting with 20-100 images and saving at regular intervals. Once you get a feel for it, you can increase the number of images. Also, use a lot of validation images to understand where it's good and where it struggles.

Analyzing the loss data can also be hard. Initially, I noticed that the batch normalization makes the pictures sepia looking. Then it needs additional training to create colors. The loss curve can be misleading because of this.

For better results, I'd recommend adding a weighted classification, to favor vibrant colors. If I were to redo it today, I'd experiment with the pix2pixHD GAN structure: https://github.com/NVIDIA/pix2pixHD

from coloring-greyscale-images.

Thanks for your relay, I will remove batch normalization and try again
I want to write an example for DL beginners, so the procedure should be easy, can be run on CPU fastly

from coloring-greyscale-images.