用 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=<輸出圖片文件路徑>
這樣就能獲得了一張藝術風格賀卡,是不是很容易?
— 完 —