HyperAIHyperAI

Command Palette

Search for a command to run...

Ajout de Contrôle Conditionnel aux Modèles de Diffusion Texte-Image

Lvmin Zhang Anyi Rao Maneesh Agrawala

Résumé

Nous présentons ControlNet, une architecture de réseau neuronal permettant d'ajouter des contrôles de conditionnement spatiaux à des modèles de diffusion d'images préentraînés et de grande taille. ControlNet verrouille les modèles de diffusion prêts pour la production et réutilise leurs couches d'encodage profondes et robustes, préentraînées avec des milliards d'images, comme un puissant squelette pour apprendre un ensemble diversifié de contrôles conditionnels. L'architecture neuronale est connectée par des "convolutions nulles" (couches de convolution initialisées à zéro) qui augmentent progressivement les paramètres à partir de zéro et garantissent que aucun bruit nuisible ne puisse affecter le fine-tuning. Nous testons divers contrôles de conditionnement, tels que les contours, la profondeur, la segmentation, la posture humaine, etc., en utilisant Stable Diffusion avec une seule ou plusieurs conditions, avec ou sans prompts. Nous montrons que l'entraînement des ControlNets est robuste avec des jeux de données petits (<50k) et grands (>1m). Des résultats exhaustifs démontrent que ControlNet pourrait faciliter une utilisation plus large pour contrôler les modèles de diffusion d'images.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
Ajout de Contrôle Conditionnel aux Modèles de Diffusion Texte-Image | Articles | HyperAI