face parsing.PyTorch

face-parsing.pytorch

コンテンツ

• トレーニング
• デモ
• 参照

トレーニング

1. トレーニングデータの準備：-Celebamask-HQデータセットをダウンロードします

   prepropess_data.pyのファイルパスを変更して実行します

python prepropess_data.py

2. Celebamask-HQデータセットを使用してモデルをトレーニングします。トレインスクリプトを実行するだけです。

    $ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py

モデルをトレーニングしたくない場合は、事前に訓練されたモデルをダウンロードして、 res/cpに保存できます。

デモ

1. 使用して訓練されたモデルを評価します。

 # evaluate using GPU
python test.py

解析マップを使用したフェイスメイク

face-makeup.pytorch

	髪	リップ
元の入力
色

参照

• ビセネット