mask prediction module network

[kukuhaza]

你好，我可以问一下 预测mask时候用的pixel-wise MLP 的网络结构吗？ 要是mask预测的训练代码也会更新的话就更好了。谢谢~~

[1073521013]

预测mask是用的pixel-wise MLP网络结构，具体可以参考Sketch-Guided Text-to-Image Diffusion Models这篇文章。mask预测的训练代码我们后续会和所有训练代码一起开源

[kukuhaza]

好的 谢谢~ 会持续关注的！