Cobra: modelo de IA de código abierto para colorear bocetos, especialmente coloreado de manga 59

小编 ·

Cobra: modelo de IA de código abierto para colorear bocetos, especialmente coloreado de manga

Cobra es un nuevo marco eficiente de preservación de ID a grano fino y largo contexto para colorear bocetos, especialmente adecuado para manga. Convierte bocetos en blanco y negro en ilustraciones a color integrando un amplio conjunto de imágenes de referencia contextuales. Cobra admite más de 200 imágenes de referencia simultáneamente, manteniendo baja latencia y ofreciendo un control flexible al usuario.

Cobra: modelo de IA de código abierto para colorear bocetos, especialmente coloreado de manga

¿Qué es Cobra?

Cobra es un nuevo y eficiente marco de preservación de ID de grano fino y largo contexto para coloreado de dibujos lineales, especialmente adecuado para el coloreado de cómics. Cobra transforma dibujos lineales en blanco y negro en ilustraciones en color mediante la integración de un amplio contexto de imágenes de referencia. Este método puede admitir más de 200 imágenes de referencia, manteniendo al mismo tiempo una baja latencia y ofreciendo un control flexible por parte del usuario.

Cobra: modelo de coloreado con IA de código abierto, para coloreado de dibujos lineales, especialmente adecuado para el coloreado de cómics. .webp

Método Cobra

  • Arquitectura principal: Cobra se basa en la arquitectura Causal Sparse DiT, que aprovecha codificaciones posicionales especialmente diseñadas, atención dispersa causal y caché de clave-valor (KV-Cache) para gestionar eficazmente referencias de largo contexto, garantizando la consistencia de los ID de color.

  • Codificación posicional reutilizable localmente: garantiza una relación de aspecto y resolución adecuadas.

  • Atención dispersa causal: reduce la complejidad computacional al disminuir los cálculos por pares entre imágenes de referencia, combinándose con el KV-Cache para mejorar la eficiencia computacional.

Resultados experimentales de Cobra

  • Resultados cualitativos: Cobra puede preservar con precisión los ID de color en escenas complejas y ofrece resultados de coloreado de alta calidad.

  • Comparación cuantitativa: en comparación con los métodos existentes de coloreado de cómics (como las métricas CLIP-IS, FID, PSNR, SSIM y AS), Cobra destaca en el coloreado de dibujos lineales y de dibujos lineales con sombras.

  • Influencia del número de imágenes de referencia: el rendimiento de Cobra sigue mejorando a medida que aumenta el número de imágenes de referencia.

  • Eficiencia del tiempo de inferencia: la atención dispersa causal (combinada con KV-Cache) demuestra mayor eficiencia con diferentes cantidades de imágenes de referencia.

Cobra: modelo de coloreado con IA de código abierto, para coloreado de dibujos lineales, especialmente adecuado para el coloreado de cómics. .webp

Escenarios de aplicación de Cobra

  • Coloreado interactivo de dibujos lineales: los usuarios pueden controlar de forma flexible los resultados del coloreado mediante indicaciones de color.

  • Coloreado de dibujos lineales: adecuado para dibujos lineales sencillos.

  • Coloreado de dibujos lineales con sombras: mantiene la fidelidad del color y los detalles en escenas complejas.

  • Coloreado de vídeos de dibujos lineales: admite el coloreado de contenido dinámico.

Preguntas frecuentes sobre Cobra

Pregunta: ¿En qué aspectos se reflejan principalmente las ventajas de Cobra en comparación con otros métodos de coloreado de dibujos lineales?

Respuesta: Las ventajas de Cobra se reflejan tanto a nivel cualitativo como cuantitativo. Cualitativamente, puede preservar con precisión los ID de color en escenas complejas, manteniendo la fidelidad del color y mejorando los detalles al procesar dibujos lineales con sombras; cuantitativamente, rinde mejor en métricas como CLIP-IS, FID, PSNR, SSIM y AS, con una mayor eficiencia en el tiempo de inferencia.

Pregunta: ¿Cómo mejora la eficiencia computacional la arquitectura DiT dispersa causal en Cobra?

Respuesta: La arquitectura DiT dispersa causal mejora la eficiencia computacional al reducir los cálculos por pares entre imágenes de referencia, aplicar atención causal unidireccional y utilizar caché de KV. La reducción de los cálculos por pares disminuye la complejidad computacional, mientras que la atención causal unidireccional y la caché de KV garantizan una transmisión eficaz de la información clave de los ID de color junto con un cálculo eficiente.

Pregunta: ¿Qué impacto tiene el número de imágenes de referencia en el rendimiento de Cobra?

Respuesta: A medida que aumenta el número de imágenes de referencia, el rendimiento de Cobra muestra una tendencia de mejora continua. Por ejemplo, la métrica CLIP-IS aumenta de 0,9083 con 4 imágenes de referencia a 0,9183 con 36; la métrica FID disminuye de 23,18 a 20,64, y métricas como PSNR, SSIM y AS también muestran mejoras correspondientes.

Enlaces relacionados

Página del proyecto:https://zhuang2002.github.io/Cobra/

Github:https://github.com/Zhuang2002/Cobra

Artículo:https://arxiv.org/abs/2504.12240