ByteDance publie InfiniteYou en open source : un framework de génération d'images basé sur FLUX préservant l'identité 133

小编 · 2025-03-23

InfiniteYou est un puissant framework de génération d'images publié par ByteDance, basé sur FLUX, qui génère diverses images tout en préservant la cohérence de l'identité. En exploitant des techniques de transformeurs de diffusion (DiTs) telles que FLUX, il résout les problèmes des méthodes existantes, notamment une similarité d'identité insuffisante, un mauvais alignement texte-image et une faible qualité de génération.

ByteDance publie InfiniteYou en open source : un framework de génération d'images basé sur FLUX préservant l'identité

Qu'est-ce qu'InfiniteYou ?

InfiniteYou est un puissant framework de génération d'images lancé par ByteDance. Basé sur FLUX, il est capable de générer diverses images tout en préservant la cohérence des personnages. Grâce à la technologie des transformateurs de diffusion (DiTs), comme FLUX, il résout les problèmes des méthodes existantes, tels qu'une similarité d'identité insuffisante, un mauvais alignement entre le texte et l'image, ainsi qu'une qualité de génération médiocre.

字节开源InfiniteYou：一种基于FLUX，能够生成各种图像的同时并保持人物一致性.jpg

Fonctionnalités principales d'InfiniteYou

Préservation de l'identité : grâce au composant InfuseNet, les caractéristiques d'identité sont injectées dans le modèle de diffusion, garantissant que l'identité du personnage dans l'image générée reste très proche de celle de l'image de référence en entrée.
Génération d'images de haute qualité : une stratégie d'entraînement en plusieurs étapes est adoptée, comprenant un pré-entraînement puis un affinage supervisé (SFT) à l'aide de données synthétiques mono-personne multi-échantillons (SPMS), ce qui améliore considérablement la qualité des images et leur rendu esthétique.
Pilotage flexible par le texte : les utilisateurs peuvent personnaliser le contenu des images générées via des descriptions en langage naturel, en prenant en charge diverses descriptions de scènes, de styles et de détails.
Conception plug-and-play : InfiniteYou est compatible avec de nombreux outils IA et frameworks existants, comme les différentes variantes de FLUX, ControlNets et LoRAs, facilitant ainsi l'extension et la personnalisation par les développeurs.

Technologies clés d'InfiniteYou

InfuseNet : il s'agit d'un composant innovant qui injecte les caractéristiques d'identité individuelles dans le modèle de génération d'images via des connexions résiduelles, garantissant que les images générées sont non seulement conformes à la description, mais préservent également les traits du visage uniques.
Stratégie d'entraînement en plusieurs étapes : le modèle est d'abord pré-entraîné sur un grand volume de données réelles, puis affiné de manière supervisée à l'aide de données synthétiques mono-personne multi-échantillons afin d'optimiser les résultats de génération et d'améliorer l'alignement entre le texte et l'image.

字节开源InfiniteYou：一种基于FLUX，能够生成各种图像的同时并保持人物一致性.webp

Cas d'utilisation d'InfiniteYou

Création de personnages virtuels : aide les utilisateurs à générer rapidement des personnages virtuels personnalisés pour des scénarios tels que les réseaux sociaux virtuels, les jeux ou les réunions virtuelles.
Avatars personnalisés et réseaux sociaux : les utilisateurs peuvent utiliser InfiniteYou pour générer des avatars personnalisés dans divers styles, enrichissant ainsi l'expérience de personnalisation sur les réseaux sociaux.
Publicité et marketing : les annonceurs peuvent exploiter InfiniteYou pour générer des images publicitaires ressemblant à leur public cible, augmentant ainsi l'attrait et le taux de conversion des publicités.
Développement de jeux : aide les développeurs à générer rapidement des modèles de personnages variés, renforçant ainsi l'immersion et le plaisir de jeu.
Création artistique : les artistes peuvent utiliser InfiniteYou pour explorer différents styles artistiques et sources d'inspiration, créant ainsi des œuvres uniques.

Comment utiliser InfiniteYou

1. Préparation de l'environnement

Configuration matérielle : il est recommandé d'utiliser un GPU hautes performances, tel qu'une NVIDIA RTX série 30 ou supérieure.
Environnement logiciel : Python (version 3.8 ou supérieure recommandée) et le framework PyTorch doivent être installés.

2. Étapes d'installation

Cloner le dépôt :

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

Installer les dépendances :

pip install -r requirements.txt

3. Génération d'images

Exécuter le script de génération :

python generate.py --prompt "描述文本" --image_path "输入图片路径" --output_path "输出图片路径"

Où :

prompt : la description textuelle en entrée, utilisée pour définir le contenu de l'image générée.
image_path : le chemin de l'image de référence en entrée, utilisé pour extraire les caractéristiques d'identité.
output_path : le chemin de sauvegarde de l'image générée.

Exemple : Supposons que vous disposiez d'une photo de vous-même et que vous souhaitiez générer une image de vous en vacances à la mer :

python generate.py --prompt "一个人在海边度假，穿着休闲装，阳光明媚" --image_path "./input.jpg" --output_path "./output.jpg"

Après exécution, InfiniteYou générera une image correspondant à votre description et à votre photo en entrée.

Les résultats montrent qu'InfiniteYou surpasse plusieurs technologies de référence actuelles, telles que PuLID-FLUX et FLUX.1-dev IP-Adapter, en termes de fidélité d'identité, de correspondance avec le texte et de qualité d'image.