face parsing.PyTorch

parpsing.pytorch

Contenido

• Capacitación
• Manifestación
• Referencias

Capacitación

1. Prepare datos de capacitación:-Descargar CelebAmask-HQ DataSet

   - Cambiar la ruta de archivo en prepropess_data.py y ejecutar

python prepropess_data.py

2. Entrena el modelo usando el conjunto de datos Celebamask-HQ: simplemente ejecute el script del tren:

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

Si no desea capacitar al modelo, puede descargar nuestro modelo previamente capacitado y guardarlo en res/cp .

Manifestación

1. Evaluar el modelo capacitado utilizando:

 # evaluate using GPU
python test.py

Maquillaje facial usando mapas de análisis

cara-makeup.pytorch

	Cabello	Labio
Entrada original
Color

Referencias

• Bisenet