本文經AI新媒體量子位（公眾號ID:QbitAI）授權轉載，轉載請聯系出處。

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

用PS嗎？

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

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

效果是這樣的：

以及這樣的：

非常干凈清晰有沒有！

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

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

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

• Python 3 (推薦使用Anaconda)
• CPU or NVIDIA GPU + CUDA CuDNN
• Pillow, 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團隊，成為一名研究科學家。

感興趣的同學戳鏈接嘗試吧：https://github.com/Mukosame/Anime2Sketch