Skip to content
Proyectos

zllrunning/face-parsing.PyTorch

Ver en GitHub

zllrunning/face-parsing.PyTorch has 2,592 GitHub stars and 497 forks. GitHot measured 0 stars in 24 hours and -1 in 7 days. Last checked 2026-11-11 08:58 UTC. Last push 2023-05-21.

Cómo se miden estas cifras

Using modified BiSeNet for face parsing in PyTorch

2,592 estrellas497 forks

Crecimiento de estrellas medido

in 24h
0
in 7 days
-1
in 15 days
Recopilando historial
in 1 month
Recopilando historial

Cambios netos en períodos que se solapan, no totales diarios. Los datos no disponibles se dejan en blanco. Las estrellas indican interés, no calidad.

README.md

Original

face-parsing.PyTorch

Contents

Training

  1. Prepare training data: -- download CelebAMask-HQ dataset

    -- change file path in the prepropess_data.py and run

python prepropess_data.py
  1. Train the model using CelebAMask-HQ dataset: Just run the train script:
    $ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py

If you do not wish to train the model, you can download our pre-trained model and save it in res/cp.

Demo

  1. Evaluate the trained model using:
# evaluate using GPU
python test.py

Face makeup using parsing maps

face-makeup.PyTorch

  Hair Lip
Original Input Original Input Original Input
Color Color Color

References

Explorar proyectos relacionados

tensorflow/tensorflow

Un framework de código abierto de aprendizaje automático para todos

201,505 estrellas

huggingface/transformers

🤗 Transformers: el framework de definición de modelos para modelos de aprendizaje automático de vanguardia en texto, visión, audio y multimodal, tanto para inferencia como para entrenamiento.

168,815 estrellas

AUTOMATIC1111/stable-diffusion-webui

Interfaz web de Stable Diffusion

165,289 estrellas

Comfy-Org/ComfyUI

La GUI, la API y el backend más potentes y modulares para modelos de difusión, con una interfaz de grafos y nodos. El motor de inferencia local más rápido del mundo.

146,271 estrellas