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

Question about training and demo #91

Closed YuMeng2v closed 2 years ago

YuMeng2v commented 2 years ago

In colorization-pytorch, the siggraph generator model forward function is written without [normalization]. But the demo code seems to be the other way, it normalize the data. Anyone who has test which is the real training way?