InfiniteYou: una herramienta de generación y edición de fotos que conserva los rasgos faciales

Introducción general

InfiniteYou es un proyecto de código abierto desarrollado por el equipo de creación inteligente ByteDance. Se basa en la tecnología de los Transformadores de Difusión (DiTs), utilizando la FLUXLa función principal del modelo .1-dev es permitir a los usuarios subir una foto e introducir una descripción de texto para generar una nueva imagen preservando la identidad de la persona. El proyecto utiliza la tecnología InfuseNet para mejorar la similitud de las identidades, combinada con un entrenamiento en varias fases para optimizar la calidad de la imagen y la alineación del texto.InfiniteYou se publicó en marzo de 2025 con código, modelos y una demo en línea, y ha recibido la atención de la comunidad técnica. Admite múltiples plug-ins y es fácil de usar para desarrolladores, investigadores y usuarios en general.

Lista de funciones

• Remodelación de la retención de identidad: Cargue una foto y una descripción de texto para generar una nueva imagen y mantener los rasgos faciales de la persona.
• Generación de imágenes de alta calidad: Emite imágenes nítidas y reduce el desenfoque, la distorsión de la mano y otros problemas.
• Optimización de la alineación del texto: Genera resultados muy coherentes con el contenido de la descripción para evitar sesgos.
• Selección de modelos: Provided aes_stage2(estética preferida) y sim_stage1(Identidad primero) dos modos.
• Extensiones Plug-inControlNet, LoRA, IP-Adapter, etc. para aumentar la flexibilidad de generación.

Utilizar la ayuda

Proceso de instalación

InfiniteYou requiere una instalación local para funcionar. A continuación se detallan los pasos:

1. Preparación medioambiental
   • Asegúrese de que Python 3.8 o posterior está instalado en su sistema.
   • Instala Git para descargar código.
   • Se recomiendan las GPU NVIDIA y CUDA para mejorar la velocidad de generación.
2. Clonación de código
   Introdúcelo en el terminal:

git clone https://github.com/bytedance/InfiniteYou.git

Ir al catálogo:

cd InfiniteYou

3. Instalación de dependencias
   Ejecute el siguiente comando para instalar las bibliotecas necesarias:

pip install -r requirements.txt

Si utiliza una GPU, deberá instalar la versión correspondiente de PyTorch, por ejemplo:

pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu124

4. Descargar modelos
   Visite https://huggingface.co/ByteDance/InfiniteYou para descargar infu_flux_v1.0 en la carpeta aes_stage2 tal vez sim_stage1 en la ubicación correspondiente del directorio del proyecto.
5. Demostración
   Instala Gradio:

pip install gradio

Inicie la interfaz local:

python app.py

Se abrirá el navegador http://127.0.0.1:7860A continuación se ofrece una lista de los productos más populares y demandados del mercado.

Funciones principales

Preservación de la identidad Reconstrucción fotográfica

1. Subir fotos
   En la interfaz de Gradio, haz clic en "Cargar imagen de identidad" y selecciona una foto nítida de tu cara.
2. Entrada Descripción
   Introduzca una descripción específica en la casilla "Prompt text", por ejemplo: "Una mujer con un vestido rojo en el bosque".
3. Parámetros de ajuste

• Selección de modelo:aes_stage2 Centrarse en la estética.sim_stage1 Énfasis en la similitud de identidades.
• Ajuste de parámetros: por defecto infusenet_conditioning_scale debido a 1.0(matemáticas) géneroinfusenet_guidance_start debido a 0.0. Si las identidades no son lo suficientemente similares, pruebe a ajustar esta última a la 0.1.
• Valor semilla (Seed): mantener el valor por defecto o establecerlo manualmente.

4. Generación de imágenes
   Haga clic en "Generar" y espere unos 120 segundos a que aparezca el resultado en la pantalla.

Razonamiento rápido de guiones

Se ejecuta en el terminal:

python test.py --id_image ./assets/examples/yann-lecun_resize.jpg --prompt "一个男人，肖像，电影风格" --out_results_dir ./results

Los resultados generados se guardan en el archivo results Carpeta.

Demostración en línea

Visite https://huggingface.co/spaces/ByteDance/InfiniteYou-FLUX para probarlo sin instalación.

Uso de enchufes

• ControlNetCarga una imagen de referencia de pose para controlar la acción que genera el resultado.
• LoRA: Se admiten los plug-ins Realismo y Antidesenfoque, y la ruta debe especificarse manualmente, por ejemplo. <path_to_lora>.
• Adaptador IPAñadir una tabla de referencia de estilo para personalizar el estilismo.

Ejemplo de funcionamiento

Quiere generar una imagen de un "hombre con traje en una sala de conferencias":

1. Sube una foto del hombre.
2. Introduce una descripción: "Un hombre trajeado en una sala de reuniones".
3. opción aes_stage2Haga clic en "Generar".
4. Comprueba los resultados y añade las palabras "un hombre" si necesitas ajustar el género.

advertencia

• Las fotografías deben ser claras en el anverso y evitar el bloqueo.
• Sea específico en sus descripciones y evite palabras vagas como "hermoso".
• El tiempo de generación varía según el hardware y puede llegar a ser de 30-60 segundos en el caso de las GPU.

escenario de aplicación

1. Contenido de las redes sociales
   Los usuarios suben un selfie y escriben "con ropa de gimnasia en el gimnasio" para generar una foto de temática fitness y compartirla.
2. arte
   El artista carga un retrato que representa a un "caballero en traje medieval" para generar un diseño conceptual.
3. prueba de investigación
   Los investigadores utilizaron InfiniteYou para comparar los efectos de la retención de identidad y validar el rendimiento de los modelos generados.

CONTROL DE CALIDAD

1. ¿Es gratis?
   Sí. El código y el modelo son de código abierto y de uso gratuito para la investigación académica.
2. ¿Cuánto tarda en generarse?
   120 segundos de media, más rápido con la GPU.
3. ¿Admite fotos multijugador?
   Actualmente optimizado para un solo jugador, el soporte multijugador necesita más desarrollo.