你见过火系的暴鲤龙吗?这个项目利用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

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

原文发布于微信公众号 - 机器之心(almosthuman2014)

原文发表时间:2019-06-05

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券