ColorFlow : ombrage de dessins animés, ombrage automatique des images en noir et blanc pour améliorer la cohérence et la qualité des couleurs de l'image

Introduction générale

ColorFlow est un outil d'auto-coloration de séquences d'images développé par l'équipe ARC de Tencent pour résoudre le problème de l'auto-coloration de séquences d'images en noir et blanc. L'outil utilise un pipeline d'ombrage amélioré par récupération pour générer avec précision les couleurs de divers éléments, y compris la couleur des cheveux et des vêtements du personnage, à partir d'un ensemble d'images de référence, en veillant à ce que les couleurs soient cohérentes avec les images de référence.ColorFlow utilise un cadre de modélisation de la diffusion en trois phases combiné à un mécanisme d'auto-attention pour obtenir un ombrage d'image de haute qualité et une cohérence des couleurs. Cet outil est particulièrement adapté aux applications industrielles dans les séries de dessins animés ou de bandes dessinées, car il améliore considérablement la contrôlabilité et la cohérence de l'identité de la coloration de l'image.

Liste des fonctions

• Récupérer les tuyaux de coloration améliorésLes couleurs sont recherchées et comparées à partir d'un ensemble d'images de référence afin de garantir la cohérence des couleurs.
• Contexte Coloration des tuyauxLes enfants de moins de 18 ans doivent être en mesure d'apprendre à colorier des images en utilisant des informations contextuelles pour maintenir l'identité cohérente des personnages et des objets.
• Pipeline guidé de super résolution: Augmente la résolution de l'image et améliore les détails et la qualité de l'image.
• Mécanisme d'auto-attentionLe projet de loi sur la protection de l'environnement a été adopté par le Parlement européen en décembre 2008. Il s'agit d'un projet de loi sur la protection de l'environnement.
• Bancs d'essai ColorFlowLe modèle d'évaluation des performances : Il fournit des références complètes pour évaluer les performances du modèle.

Utiliser l'aide

Processus d'installation

1. entrepôt de clones: :

   git clone https://github.com/TencentARC/ColorFlow
cd ColorFlow

2. Configuration de l'environnement PythonPour cela, il faut : s'assurer qu'Anaconda ou Miniconda est installé, puis créer et activer un environnement Python et installer les dépendances nécessaires :

   conda create -n colorflow python=3.8.5
conda activate colorflow
pip install -r requirements.txt

3. Exécution de l'applicationLancer l'interface de Gradio :

   python app.py

4. Accès à ColorFlowOuvrez http://localhost:7860 dans votre navigateur. Si vous exécutez l'application sur un serveur distant, remplacez localhost par l'adresse IP ou le nom de domaine du serveur.

Fonction Opération Déroulement

1. Récupérer le coloriage amélioré: :
   • Téléchargez un ensemble d'images de référence et ColorFlow récupérera automatiquement les couleurs et les fera correspondre.
   • Sélectionnez la séquence d'images en noir et blanc à colorier et le système les colorise en fonction du pool d'images de référence.
2. coloration contextuelle: :
   • Le système analyse automatiquement les informations contextuelles de la séquence d'images afin de garantir l'identité cohérente des personnages et des objets.
   • Les utilisateurs peuvent ajuster manuellement les paramètres de correspondance des couleurs pour obtenir les meilleurs résultats d'ombrage.
3. Super-résolution d'amorçage: :
   • Une fois la coloration terminée, le système effectue automatiquement un traitement de super-résolution pour améliorer les détails et la qualité de l'image.
   • Les utilisateurs peuvent sélectionner différents niveaux de super résolution et ajuster la qualité de l'image en fonction de leurs besoins.

Mode d'emploi détaillé

• Télécharger une image de référenceCliquez sur le bouton "Upload", sélectionnez le fichier de l'image de référence et le système effectuera automatiquement une recherche de couleurs.
• Sélection d'une séquence d'images en noir et blancCliquez sur le bouton "Sélectionner un fichier" pour télécharger une séquence d'images en noir et blanc à colorier.
• Réglage des paramètres de correspondance des couleursDans l'interface "Parameter Setting", les utilisateurs peuvent régler les paramètres pertinents de la correspondance des couleurs, tels que l'intensité des couleurs, la précision de la correspondance, etc.
• Voir les résultats du coloriageCliquez sur le bouton "Démarrer la coloration", le système effectuera automatiquement le processus de coloration et l'utilisateur pourra visualiser les résultats de la coloration en temps réel.
• Enregistrer l'image à colorierLorsque le coloriage est terminé, cliquez sur le bouton "Enregistrer" pour sauvegarder l'image colorée localement.