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

	Волосы	Губа
Исходный ввод
Цвет

Ссылки

• Бисенет