不裝了，我攤牌了！辰哥實現了人像卡通化的功能，玩了絕對上癮！
今天咱們來搞點有趣的事情：人像卡通化。本文的人像卡通化功能沒有使用第三方接口功能，辰哥知道百度有提供接口，並且十來行代碼就可以搞定，但是有使用次數限制，所以辰哥就通過搭建神經網絡模型，借助數據集進行訓練，最后得到模型。
這樣得到的人像卡通化的功能可以順便使用，並且還可以通過提高數據集的質量或者調整參數，讓生成的卡通人像更加逼真！
先來看一下效果：
看完效果之后，是不是覺得很神奇？辰哥告訴你，實際很簡單，看完本文之后，你也可以自己動手直接生成你自己喜歡的動漫頭像。
01、搭建環境
這里辰哥使用的是開源平台github的源碼，里面有完整的模型結構、模型文件、數據集等。項目地址在下方
項目如下：
https://github.com/minivision-ai/photo2cartoon
可能看到這篇文章的讀者沒有深度學習的基礎，但是沒關系，辰哥一步一步教你如何將這個項目搭建起來，並且生成自己的卡通頭像！
1.安裝庫
大家下載好源碼之后，在運行之前，先把運行的環境搭建一下。
上面是項目中需要安裝的庫，實際上通過下面的四個命令安裝即可（其中項目tensorflow-gpu說需要在gpu顯卡下運行，但是實際上在咱們自己的筆記本或者電腦上運行也可以）
pip install onnxruntime
pip install face-alignment
pip install pytorch
pip install tensorflow==1.15
這里需要安裝tensorflow版本是1，非2，不然就出現下面的報錯（原因：1和2這兩個版本很很大區別）
2.下載模型和數據集
從github上下載好代碼之后，其目錄結構如下：
這里給大家簡單介紹一下相關文件夾和文件作用是什么？
文件夾：
dataset：存放訓練數據集
images：存放測試數據集（預測模型效果所使用的圖片存放文件夾）
models：訓練好的模型存放在該目錄下
utils：里面存放一些處理圖片，模型結構等py文件
Py文件:
train.py：訓練模型
test.py：測試模型（生成一張動漫人像圖片）
主要就是這兩個py文件需要去了解，即如何訓練模型和使用模型（生成動漫人像圖）
在項目中已提供了訓練好的模型和數據集
在項目中還沒有包含這些文件（已訓練好模型、數據集等），需要我們再去下載，通過上面的下載地址，下載好之后再放入對應的文件夾即可。
這樣我們就將這個項目的環境搭建完畢！
02、生成人像動漫圖
1.訓練模型
（已下載好了訓練模型，想直接使用的可以跳過這一步，直接進行測試生成人像動漫圖）
運行環境搭建完畢之后，下面可以開始進行訓練模型。
先是對數據集進行預處理：
python data_process.py --data_path YourPhotoFolderPath --save_path YourSaveFolderPath
然后開始訓練
python train.py --dataset photo2cartoon
如果是下載了預訓練模型后，也可以在預訓練模型的基礎上進行訓練
python train.py --dataset photo2cartoon --pretrained_weights models/photo2cartoon_weights.pt
訓練結束之后的模型保存到models文件夾中。
2.測試生成人像動漫圖
將需要生成動漫人像的原圖放到images下：
執行生成命令：
# 使用默認模型
python test.py --photo_path ./images/lx.jpg --save_path ./images/cartoon_lx.png
# 使用onnx模型
python test_onnx.py --photo_path ./images/lx.jpg --save_path ./images/cartoon_lx.png
項目中提供了兩種模型，因此在執行生成命令時也有兩個（選擇其中一條執行即可），其中原圖片是：lx.jpg，生成的動漫人像是：cartoon_lx.png。
運行結果如下：
可以看到由一張真人人像圖成功的生成了動漫人像圖。
整個過程還是比較簡單的，沒有學過深度學習的小伙伴也可以將代碼跑起來！！
03、小結
本文中教會了大家如何通過一張真人圖去生成對應的動漫圖，在文中也詳細地講解了如何去搭建環境，並代碼跑起來。
鑒於可能很多讀者不是學習深度學習方向的，所以文中就不太細於講解代碼，只是教會大家如何將代碼跑起來，感興趣的讀者可以自己嘗試！
一定要動手嘗試 ！一定要動手嘗試 ！一定要動手嘗試！