見過火系的暴鯉龍嗎?這個項目利用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
※面對危機再次出手!華為耗資5千萬美金買下俄羅斯人臉識別公司
※解讀目標檢測新範式:Segmentations is All You Need
TAG:機器之心 |