material editing unet 源码

材料编辑网 使用U-net架构的基于学习的材料编辑。 网络估算输入图像的材质,并输出带有目标材质的编辑图像。 还支持漫反射和镜面反射,并且仅可以更改材质而无需更改其他属性(形状和照明)。 这是一种受监督的方法,需要目标材料和其他材料的图像对进行训练。 准备数据 您可以自己准备训练和测试数据,还可以通过Mitsuba渲染器和ShapeNet(3D模型数据库)生成图像。 它规范化ShapeNet的形状,并使用您选择的不同材质自动渲染形状。 训练