参与:路作者:Riley Wong

你见过火系的暴鲤龙吗?这个项目利用CycleGAN生成不同属性的神奇宝贝

水系暴鲤龙还可以变成火系、草系、电系?最近研究者 Riley Wong 做了一个项目,他训练了一个可以改变神奇宝贝属性的 CycleGAN

了解属性相克,可以帮助训练师们更好地战斗。可是如果神奇宝贝的属性变了呢?有 reddit 评论表示:很想在游戏里见到不同属性的神奇宝贝~以及很好奇改变属性后的皮卡丘会是什么样。

先看一下水系暴鲤龙和火系凤王「变身」后的结果:

水系暴鲤龙、火系暴鲤龙、草系暴鲤龙、电系暴鲤龙齐聚一堂……

火系凤王(Ho-oh)vs 恶系凤王

模型

CycleGAN(ICCV 2017)变革了基于图像的计算机图形学,可作为一种通用框架将一组图像中的视觉风格迁移到其它图像。例如,将夏天转化为冬天、将马转换为斑马及利用计算机图形渲染生成真实图像等。

Riley Wong 利用 CycleGAN 框架执行该项目,他训练模型将不同属性的神奇宝贝图像风格相互转换,如将水系转换为火系。

他使用 PyTorch 实现该项目,GitHub 地址:https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

训练数据

Riley Wong 写了一个脚本,将包含 1-7 代神奇宝贝的原始数据集按照主要属性进行分类。

  • 原始神奇宝贝图像数据集地址:https://www.kaggle.com/vishalsubbiah/pokemon-images-and-types

  • 脚本地址:https://github.com/rileynwong/sort-pokemon-images-by-type

  • 按属性分类后的神奇宝贝图像数据集地址:https://github.com/rileynwong/pokemon-images-dataset-by-type

效果

下面展示了更多属性转换效果,左侧为神奇宝贝原始图像,右侧为风格迁移后的版本。

水属性 → 其他属性

火属性 → 其他属性

草属性 → 其他属性

电属性 → 其他属性

恶属性 → 其他属性

参考链接:https://www.rileynwong.com/blog/2019/5/22/pokemon2pokemon-using-cyclegan-to-generate-pokemon-as-different-elemental-types

工程生成对抗网络生成模型计算机视觉
相关数据
计算机图形技术

图像数据处理、计算机图像(英语:Computer Graphics)是指用计算机所创造的图形。更具体的说,就是在计算机上用专门的软件和硬件用来表现和控制图像数据。

CycleGAN技术

GAN的一个变种

推荐文章
暂无评论
暂无评论~