曾在位元組實習的程式設計師小姐姐，教你一步提取動漫線稿！比PS更清晰

2021-05-26 17:00:07

豐色發自凹非寺量子位報道 | 公眾號 QbitAI

每當你看到喜歡的動漫作品，想把線稿提出來、進行重新上色等加工時，你是怎麼做的？

用PS嗎？

是不是有點麻煩？結果還很糊？

圖片

GitHub上有一個小姐姐做了一個動漫線稿提取器，只需一行程式碼就能搞定！

效果是這樣的：

圖片

以及這樣的：

圖片

非常乾淨清晰有沒有！

如何一步搞定動漫線稿提取

這款「又快又好」的線稿提取器名叫Anime2Sketch，可用於插畫、動漫藝術和漫畫。

使用Python語言開發，使用前確保具備以下條件：

Python 3 (推薦使用Anaconda)CPU or NVIDIA GPU + CUDA CuDNNPillow, PyTorch首先執行以下兩步：

安裝所需的軟體包：pip install -r requirements.txt從GoogleDrive下載預訓練權重檔案，然後將其放入weights /資料夾中然後就可以進行測試了！使用下面這行程式碼：

python3 test.py --dataroot 你的輸入目錄 --load_size 512 --output_dir 你的輸出目錄

其中參數load_size：由於記憶體限制，程式需要在處理之前調整輸入影象的大小。預設情況下，將其大小調整為512x512。

這麼方便，怎麼實現的呢？

（小遺憾是目前只可在Linux或macOS上運行。）

基於一篇論文實現

作者介紹道，這款線稿提取器程式基於論文《用於草圖到照片合成的對抗性開放領域自適應演算法》。

圖片

也是由她作為一作，將在位元組跳動公司實習時的一部分成果寫成的論文。

論文中提到，為了從照片中合成缺少的手繪草圖，他們提出了一個從草圖到照片和從照片到草圖的聯合學習框架。

並進一步提出了一個簡單而有效的開放域取樣和優化策略。

他們的方法利用所學的域內（in-domain）資料的草圖到照片和照片到草圖的對映，可以遷移到開放域（open-domain）類中。

最終他們將這篇論文的研究成果應用到動漫線稿提取，做成了這樣一款程式。

有網友看到後很興奮：可以用來給自家娃做「神奇寶貝「和「爆旋陀螺」的塗色頁了！

圖片

有人好奇這和用Photoshop技術上有什麼不一樣？其他網友解釋道因為Photoshop僅檢測邊緣，會有「偽像」，這款明顯更乾淨！

圖片

不過有網友指出最後的圖片輸出質量還可再提高，比如使用PNG格式。(原來是JPEG)

圖片

還有人說起了題外話，對示例圖片裡的動漫海盜戰記

（Vinland Saga）

進行了瘋狂安利！

圖片

作者介紹

圖片

向曉宇（圖右），@Mukosame。普渡大學博士在讀，電氣與計算機工程，本科清華畢業。博士階段研究重點是影象和視訊增強，面部對齊和識別的深度學習。

今年即將加入Facebook Reality Labs實驗室的On-Device AI團隊，成為一名研究科學家。