search

richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 921 forks source link

Params of SIGGRAPHGenerator #81

Open MSKTHU opened 3 years ago

MSKTHU commented 3 years ago

Hi, I think you did a really greate job. Here is my one tiny question: What is the params input_B and mask_B do in SIGGRAPHGenerator.forward? Is input_B the same format as input_A? And what should mask_B like? By the way, do you have any examples on these params? Thank you very much.

richzhang commented 3 years ago

Thanks for the question. I should add some documentation and examples eventually. But it is from this paper and is better documented in this repo: https://github.com/junyanz/interactive-deep-colorization.