ColorFlow: sombreado de dibujos animados, sombreado automático de imágenes en blanco y negro para mejorar la coherencia y la calidad del color de las imágenes.

Introducción general

ColorFlow es una herramienta de autocoloreado de secuencias de imágenes desarrollada por el equipo ARC de Tencent para resolver el problema del autocoloreado de secuencias de imágenes en blanco y negro. La herramienta utiliza un proceso de sombreado mejorado por recuperación para generar con precisión los colores de varios elementos, como el color del pelo y la ropa del personaje, a partir de un conjunto de imágenes de referencia, garantizando que los colores sean coherentes con las imágenes de referencia.ColorFlow emplea un marco de modelado de difusión de tres fases combinado con un mecanismo de autoatención para lograr un sombreado de imágenes de alta calidad y la coherencia del color. La herramienta es especialmente adecuada para aplicaciones industriales en series de dibujos animados o cómics, ya que mejora notablemente la capacidad de control y la coherencia de identidad del coloreado de imágenes.

Lista de funciones

• Recuperar tubos de coloración mejoradaRecuperación y correspondencia de colores mediante un conjunto de imágenes de referencia para garantizar la coherencia cromática.
• Contexto Colorear tuberías: Coloreado de imágenes utilizando información contextual para mantener la identidad coherente de personajes y objetos.
• Superresolución guiadaAumenta la resolución de la imagen y mejora su detalle y calidad.
• Mecanismo de autoatenciónUtilización de mecanismos de autoatención en modelos de difusión para el aprendizaje robusto del contexto y la correspondencia de colores.
• Pruebas comparativas ColorFlow-Bench: Proporciona una evaluación comparativa de referencia exhaustiva para evaluar el rendimiento de los modelos.

Utilizar la ayuda

Proceso de instalación

1. almacén de clones::

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

2. Configuración del entorno PythonAsegúrese de que Anaconda o Miniconda están instalados, cree y active un entorno Python e instale las dependencias necesarias:

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

3. Ejecutar la aplicaciónInicia la interfaz de Gradio:

   python app.py

4. Acceso a ColorFlow: Abra http://localhost:7860 en su navegador. Si ejecuta la aplicación en un servidor remoto, sustituya localhost por la dirección IP o el nombre de dominio del servidor.

Función Flujo de operaciones

1. Recuperar la coloración mejorada::
   • Cargue un grupo de imágenes de referencia y ColorFlow recuperará e igualará automáticamente los colores.
   • Seleccione la secuencia de imágenes en blanco y negro que desea colorear y el sistema las igualará y coloreará según el conjunto de imágenes de referencia.
2. coloración sensible al contexto::
   • El sistema analiza automáticamente la información contextual de las secuencias de imágenes para garantizar la identidad coherente de personajes y objetos.
   • Los usuarios pueden ajustar manualmente los parámetros de igualación de colores para obtener los mejores resultados de sombreado.
3. Superresolución de Bootstrap::
   • Una vez finalizada la coloración, el sistema realiza automáticamente un procesamiento de superresolución para mejorar el detalle y la calidad de la imagen.
   • Los usuarios pueden seleccionar distintos niveles de superresolución y ajustar la calidad de imagen según sus necesidades.

Instrucciones de uso detalladas

• Cargar una imagen de referenciaHaga clic en el botón "Cargar", seleccione el archivo de imagen de referencia y el sistema realizará automáticamente la búsqueda de colores.
• Selección de una secuencia de imágenes en blanco y negroHaga clic en el botón "Seleccionar archivo" para cargar una secuencia de imágenes en blanco y negro para colorear.
• Ajuste de los parámetros de ajuste del colorEn la interfaz "Configuración de parámetros", los usuarios pueden ajustar los parámetros relacionados con la coincidencia de colores, como la intensidad del color, la precisión de la coincidencia, etc.
• Ver resultados de colorearAl hacer clic en el botón "Iniciar coloración", el sistema llevará a cabo automáticamente el proceso de coloración y el usuario podrá ver los resultados de la coloración en tiempo real.
• Guardar imagen para colorearCuando termine de colorear, haga clic en el botón "Guardar" para guardar la imagen coloreada localmente.