你见过火系的暴鲤龙吗？这个项目利用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 本文为机器之心编译，转载请联系本公众号获得授权。 ?------------------------------------------------ 加入机器之心（全职记者 / 实习生）：hr@jiqizhixin.com 投稿或寻求报道：content@jiqizhixin.com 广告 & 商务合作：bd@jiqizhixin.com