Guía de estilización de rostros

Imágenes en paralelo de un primer plano de un hombre y una versión generada de la fotografía que usa un estilo de caricatura.

La tarea Face Stylizer de MediaPipe te permite aplicar estilizaciones de rostro a los rostros de una imagen. Puedes usar esta tarea para crear avatares virtuales en varios estilos.

La tarea usa el modelo BlazeFaceStylizer, que consta de un generador de rostros y un codificador de rostros. El generador de rostros BlazeStyleGAN, que es una implementación liviana de la familia de modelos StyleGAN, genera y edita rostros según un estilo especificado. El codificador de rostros, que usa una red principal de MobileNet V2, asigna las imágenes de entrada a los rostros que genera el generador de rostros.

Pruébala.

Comenzar

Para comenzar a usar esta tarea, sigue una de estas guías de implementación para tu plataforma de destino. En estas guías específicas de la plataforma, se explica una implementación básica de esta tarea con un modelo recomendado y ejemplos de código con las opciones de configuración recomendadas:

Detalles de la tarea

En esta sección, se describen las capacidades, las entradas, las salidas y las opciones de configuración de esta tarea.

Funciones

  • Procesamiento de imágenes de entrada: El procesamiento incluye la rotación de imágenes, el cambio de tamaño, la normalización y la conversión de espacios de color.
Entradas de tareas Resultados de las tareas
Imágenes fijas Face Stylizer genera una imagen con una estilización del rostro más prominente de la imagen de entrada.

Opciones de configuración

Esta tarea tiene las siguientes opciones de configuración:

Nombre de la opción Descripción Rango de valores Valor predeterminado
errorListener Establece un objeto de escucha de errores opcional. N/A Not set

Modelos

El Face Stylizer requiere que se descargue y almacene un modelo de estilización de rostro en el directorio de tu proyecto. Los modelos de estilización de rostros de esta sección se basan en la arquitectura de BlazeStyleGAN. Cada modelo se entrenó para aplicar un estilo específico a los rostros dentro de las imágenes de entrada.

Esbozo en color

El modelo transforma los rostros en una imagen que imita un boceto con trazos de lápiz y pincel de colores. A continuación, se muestra el estilo que se usó para entrenar este modelo:

Resultado del boceto en color

Nombre del modelo Forma de entrada Tipo de cuantización Versiones
Boceto en color 256 x 256 x 3 Float32 Más reciente

Tinta de color

El modelo transforma los rostros en una imagen que imita una pintura con acuarela. A continuación, se muestra el estilo que se usó para entrenar este modelo:

Salida de tinta de color

Nombre del modelo Forma de entrada Tipo de cuantización Versiones
Tinta de color 256 x 256 x 3 Float32 Más reciente

Pintura al óleo

El modelo transforma los rostros en una imagen que imita una pintura al óleo. A continuación, se muestra el estilo que se usó para entrenar este modelo:

resultado de la pintura al óleo

Nombre del modelo Forma de entrada Tipo de cuantización Versiones
Pintura al óleo 256 x 256 x 3 Float32 Más reciente