見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

水系暴鯉龍還可以變成火系、草系、電系?最近研究者 Riley Wong 做了一個項目,他訓練了一個可以改變神奇寶貝屬性的 CycleGAN。

瞭解屬性相剋,可以幫助訓練師們更好地戰鬥。可是如果神奇寶貝的屬性變了呢?有 reddit 評論表示:很想在遊戲裡見到不同屬性的神奇寶貝~以及很好奇改變屬性後的皮卡丘會是什麼樣。

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝


先看一下水系暴鯉龍和火系鳳王「變身」後的結果:

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

水系暴鯉龍、火系暴鯉龍、草系暴鯉龍、電系暴鯉龍齊聚一堂……

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

火系鳳王(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

效果

下面展示了更多屬性轉換效果,左側為神奇寶貝原始圖像,右側為風格遷移後的版本。

水屬性 → 其他屬性

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝


見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝


火屬性 → 其他屬性

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝


見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

草屬性 → 其他屬性

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

電屬性 → 其他屬性

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝

惡屬性 → 其他屬性

見過火系的暴鯉龍嗎?這個項目利用CycleGAN生成不同屬性神奇寶貝


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

相關推薦

推薦中...