search

junyanz / interactive-deep-colorization

Deep learning software for colorizing black and white images with a few clicks.
https://richzhang.github.io/ideepcolor/
MIT License
2.69k stars 447 forks source link

Pytorch model doesn't produce the intended output #63

Open Samir55 opened 5 years ago

Samir55 commented 5 years ago

Hello authors, I am encountering a problem that I couldn’t reproduce the same output by training your model on the ILSVRC2012 training set. The model losses (G_CE And G_entr) tend to stabilize around 3.1 across epochs. Also, the colorized images tend to have a somehow yellowish color. I trained the model for 6 epochs. I am using Python 3.5 and CUDA 9.1 with GTX 1060 GPU. Can you help me discover what’s going on? I am adding here some output pictures and the loss graph. Thanks for your time and efforts.

00000195_0p031_fake_reg 00000198_0p125_fake_reg loss_graph

richzhang commented 5 years ago

I believe I did not do the initialization properly in the pytorch code. I am looking into it. Thanks for your patience.