NVIDIA 开放机器学习系统代码,可根据草图合成景观

英伟达公司 опубликовала 机器学习系统源代码 (GauGAN),它允许您根据草图以及与项目相关的草图合成真实的景观 训练有素的模型。 该系统是 证明了 2019 月份的 GTC XNUMX 大会上,但代码昨天才发布。 动态 打开 根据专有许可 CC BY-NC-SA 4.0 (知识共享署名-非商业性-相同方式共享 4.0),仅允许非商业用途。 代码是使用Python框架编写的 PyTorch.

NVIDIA 开放机器学习系统代码,可根据草图合成景观

草图以分段地图的形式绘制,确定场景中近似对象的位置。 生成的对象的性质是使用颜色标记指定的。 例如,蓝色填充变为天空,蓝色填充为水,深绿色填充为树木,浅绿色填充为草,浅棕色填充为岩石,深棕色填充为山脉,灰色填充为雪,棕色线填充为道路,蓝色填充填充为道路。线入河此外,根据参考图像的选择,确定整体构图风格和一天中的时间。 所提议的用于创建虚拟世界的工具对于从建筑师和城市规划师到游戏开发人员和景观设计师的广泛专家来说都是有用的。

NVIDIA 开放机器学习系统代码,可根据草图合成景观

对象由生成对抗神经网络合成(),它根据示意性分段地图创建逼真的图像,借用在数百万张照片上预先训练的模型中的细节。 与之前开发的图像合成系统不同,所提出的方法基于使用自适应空间变换,然后是基于机器学习的变换。 处理分段地图而不是语义标记可以让您获得精确的匹配结果并控制样式。

NVIDIA 开放机器学习系统代码,可根据草图合成景观

为了实现真实感,两个神经网络相互竞争:生成器和鉴别器。 生成器根据真实照片的混合元素生成图像,鉴别器识别与真实图像可能存在的偏差。 结果,形成了反馈,在此基础上,生成器开始组合越来越好的样本,直到鉴别器不再将它们与真实样本区分开来。

来源: opennet.ru

添加评论