來源:快學Python
最近遇到了一個問題,需要將頭像轉化動漫風,當時第一想法就是找現成的輪子。比如我們在之前的文章中《超有趣的Python實戰:將圖片轉成像素風!》,就是通過調用名為Tiler
的開源庫,生成可愛像素風圖片!
將頭像轉化為動漫風,就是希望在保持原圖像信息和紋理細節的同時,將真實照片轉換為動漫/卡通風格的非真實感圖像。目前除了百度api,Github上還有不少開源庫可供我們直接使用。
其中AnimeGAN是來自武漢大學和湖北工業大學的一項研究,采用的是神經風格遷移 + 生成對抗網絡(GAN)的組合,實現效果非常符合我們的需求。
AnimeGAN最早使用的是Tensorflow框架,不過查詢資料後發現該項目已支持PyTorch框架。
地址:https://github.com/bryandlee/animegan2-pytorch
而我正好之前基於PyTorch實現過微博評論情感分析,所以用起來也不會有什麼負擔,還省得安裝庫了。
PyTorch是一個開源的Python機器學習庫,基於Torch,用於自然語言處理等應用程序。這個深度學習框架,可以應用在數值建模、圖像建模、文本建模、音頻建模等等多個方向。
安裝Pytorch會比其他庫麻煩一些,如果去官方安裝下載,需要根據實際配置得到適合自己的安裝命令。
如果你通過上面這種步驟發現下載速度慢,或者遇到各種問題,不妨試試下面這個網站:
https://download.pytorch.org/whl/torch_stable.html
網站視圖:
安裝Pytorch框架後,我們便可以將該animegan2-pytorch
項目clone到本地/直接下載下來:
git clone https://github.com/bryandlee/animegan2-pytorch
下載到本地的文末目錄如下圖所示:
其中weights文件夾下包含四個權重,選擇對應的權重就能實現自己想要的動漫風格遷移。samples下的inputs文件夾中儲存了圖片示例,可以直接拿來試水。除此以外,我還在同一路徑下新建了output文件夾,用於存放處理後的圖片。
接下來,我們只需要在命令行中運行test.py
腳本就可以調用該項目,具體命令格式如下:
python test.py --checkpoint [模型文件路徑] --input_dir [輸入圖像所在目錄] --output_dir [輸出目錄] --device [設備選擇,cpu或者cuda]
實際操作:
由於是像實現人臉動漫遷移,所以我分別使用了face_paint_512_v1.pt
和face_paint_512_v2.pt
權重,效果如下圖所示:
換一張圖看一下:
個人覺得face_paint_512_v2.pt
權重的處理成品更符合我對動漫風的想象,大家怎麼看呢?
最後我也試了試paprika.pt
對風景圖片進行動漫遷移。
效果如下圖所示:
如果大家對將頭像/圖片轉化動漫風感興趣,不妨也來試試~
注意!
注意!
如果看完本文的小伙伴,不想安裝Pytorch,又想直接打造一個自己專屬的漫畫臉?
可以電腦打開這個網址:https://huggingface.co/spaces/akhaliq/AnimeGANv2,這是在線AnimeGANv2的APP,無需你本地安裝什麼框架,直接就可以進行轉換。
我們的文章到此就結束啦。