Params of SIGGRAPHGenerator

[MSKTHU]

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]

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.