FaceSwap

模型下載

模型下載連結 ( From : photo2cartoon )

下載後會有 cartoon_data 、 photo2cartoon_weights.pt 、 seg_model_384.pb 、 model_mobilefacenet.pth

1.人像卡通化預訓練模型：photo2cartoon_weights.pt，存放在models路徑下。(20200504更新)

2.頭像分割模型：seg_model_384.pb，存放在utils路徑下。

3.人臉識別預訓練模型：model_mobilefacenet.pth，存放在models路徑下。

4.人像卡通化onnx模型：photo2cartoon_weights.onnx (Google雲端硬碟)，存放在models路徑下。

卡通化開源數據：cartoon_data，包含trainB和test。

PRNet模型下載連結

5.下載後會有 ( 256_256_resfcn256_weight.data-00000-of-00001 ) 檔案，將檔案移動至 Data/net-data 資料夾裡面。

PRNet Train Data: https://drive.google.com/file/d/1UoE-XuW1SDLUjZmJPkIZ1MLxvQFgmTFH/view

https://github.com/YadiraF/PRNet/tree/master/Data

安裝環境

• python==3.6.12

• opencv-python==4.1.0.25

• PyQt5==5.15.4

• matplotlib==3.3.3

• face-alignment==1.1.1

• tensorflow-gpu==1.14.0

• torchvision==0.8.1

• dlib==19.21.0

執行檔案

python QTmainV2.py

初始畫面

圖片換臉界面

• 目前圖片支援格式 ⇒ .jpg

人臉置換

人臉置換 ( 卡通化 )

Reference

• 人像卡通化 ( Photo to Cartoon ) [GitCode]
• Joint 3D Face Reconstruction and Dense Alignment with Position Map Regression Network ( PRNet ) [GitCode] [Paper]
• 技術參考 ⇒ ( 基于PRNet的3D人脸重建与替换 )
• 範例人臉圖片來源 ( https://generated.photos/ )

PS

很多功能都不是很完全(誤)