用python將你的頭像“卡通化”

---

最近看到一個有趣的python項目，可以將一張普通人像照片快速轉成卡通風格！
這個項目讓我想起了以前玩QQ秀的時光。有了這個項目，你也可以把頭像卡通化了，再也不用找同學幫忙了。

人像卡通風格渲染的目標是，在保持原圖像ID信息和紋理細節(jié)的同時，將真實照片轉換為卡通風格的非真實感圖像。
該項目的思路是，從大量照片/卡通數據中習得照片到卡通畫的映射，然后根據這樣的映射關系，對新的照片進行轉換。


項目地址：https://github.com/minivision-ai/photo2cartoon項目所需的主要依賴庫如下：

• python 3.6

• pytorch 1.4

• tensorflow-gpu 1.14

• face-alignment

• dlib

訓練

1.數據準備訓練數據包括真實照片和卡通畫像，為降低訓練復雜度，該項目對兩類數據進行了如下預處理：

• 檢測人臉及關鍵點。

• 根據關鍵點旋轉校正人臉。

• 將關鍵點邊界框按固定的比例擴張并裁剪出人臉區(qū)域。

• 使用人像分割模型將背景置白。

項目開源了204張?zhí)幚砗蟮目ㄍó嫈祿?，您還需準備約1000張人像照片（為匹配卡通數據，盡量使用亞洲年輕女性照片，人臉大小最好超過200x200像素），使用以下命令進行預處理：

python data_process.py --data_path YourPhotoFolderPath --save_path YourSaveFolderPath

將處理后的數據按照以下層級存放，trainA、testA中存放照片頭像數據，trainB、testB中存放卡通頭像數據。

├── dataset
    └── photo2cartoon
        ├── trainA
            ├── xxx.jpg
            ├── yyy.png
            └── ...
        ├── trainB
            ├── zzz.jpg
            ├── www.png
            └── ...
        ├── testA
            ├── aaa.jpg
            ├── bbb.png
            └── ...
        └── testB
            ├── ccc.jpg
            ├── ddd.png
            └── ...

2.訓練重新訓練:

python train.py --dataset photo2cartoon

加載預訓練參數:

python train.py --dataset photo2cartoon --pretrained_weights models/photo2cartoon_weights.pt

測試

將一張測試照片轉換為卡通風格：

python test.py --photo_path ./images/photo_test.jpg --save_path ./images/cartoon_result.png

放個測試效果圖片大家感受一下（猜猜是誰）
說明下，該項目為小視科技卡通肖像探索項目?？墒褂梦⑿艗呙柘路蕉S碼或搜索“AI卡通秀”小程序體驗卡通化效果。項目地址：https://github.com/minivision-ai/photo2cartoon作者：天作?

來源：天作之程

_往期文章推薦_Python+OpenCV十幾行代碼模仿世界名畫

---