用 AI 幫你畫新年賀卡 :只需輸入幾個單詞 ,就能模仿大師名作

  • 2019 年 11 月 12 日
  • 筆記

如果你的手法拙劣,沒有任何藝術細菌,自己作畫完全無法見人。但是你想給妹子送上一張有個性的新年賀卡怎麼辦?

雖然沒有梵高、畢加索的才華,但你能讓AI學會做這件事啊!

有了這個ArtistAI,你甚至連簡筆畫都不用會,只要輕鬆地在鍵盤上敲擊幾下,就能製作一張名家風格的個性化電子賀卡。

讓大師為你畫畫

無論是什麼物體、什麼風格,ArtistAI統統能拿下。

告訴AI畫一個籃子,再給它一張表現主義大師蒙克的名畫《尖叫》,就能得到這樣一張賀卡:

又或者是你想讓大師畢加索為你畫一朵花:

都能輕鬆搞定。

從單詞到賀卡

ArtistAI內集成了NLP、影像生成、風格遷移等技術,3步內解決問題。

首先,我們必須要讓ArtistAI的腦洞足夠大,無論輸入什麼抽象單詞,總能想到要畫什麼。

為此,需要使用GloVe單詞嵌入模型,用維基百科詞條進行訓練。將輸入的單詞和Google「猜畫小歌」里的圖片分類進行比較,找出最貼近的分類。

不管你輸入什麼,ArtistAI總能聯想到一個「猜畫小歌」里合適的分類。

但有時它也不太聰明,比如輸入美國演員「Chris Pine」,居然會聯想到「樹」。所以你還是對它直接一點,別太刁難了。

第二步,讓ArtistAI向「猜畫小歌」學會塗鴉。

在上一步中,我們讓它腦洞大開想到了「樹」,接著使用名為Sketch-RNN的變分自動編碼器(VAE)從數據集學習足夠多的塗鴉,然後ArtistAI就能畫出自己的簡筆畫了。

最後一步,我們訓練ArtistAI將塗鴉變成有名家風格的畫作。

這一步中,使用了神經風格遷移技術。AI嘗試理解特定畫家的筆觸和調色風格,並將這種風格遷移到剛才的塗鴉上。

使用教程

ArtistAI目前已經在Github上開源了,雖然叫christmAIs,為聖誕賀卡而生,但是馬上新年也能用上啊。

它的安裝過程非常簡單。直接從GitHub上拉取項目,甚至可以傻瓜一鍵式安裝:

source venv/bin/activate # Highly recommended

./install-christmais.sh

通過兩行命令直接安裝好了依賴的軟體庫和風格遷移模型。

然後在終端里用Python運行,命令中各項參數如下:

python -m christmais.tasks.christmais_time

–input=<輸入某個單詞>

–style=<你喜歡的名畫所在路徑>

–output=<輸出圖片文件路徑>

這樣就能獲得了一張藝術風格賀卡,是不是很容易?