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

选自Riley Wong博客

作者：Riley Wong

机器之心编译

参与：路

水系暴鲤龙还可以变成火系、草系、电系？最近研究者 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

本文为机器之心编译，转载请联系本公众号获得授权。