DDcolor AI 阿里达摩院图片上色模型

详细介绍

DDColor 是由阿里达摩院研究的基于深度学习技术的图像上色模型，能够自动将黑白或灰度图像着色，使图像更加生动逼真。

主要功能

• 双解码器技术：DDColor 模型包括一个图像编码器和两个解码器，即图像解码器和颜色解码器。图像解码器完成视觉特征的上采样过程，而颜色解码器基于一个 transformer 进行颜色查询的解码，并使用多尺度的视觉特征帮助颜色嵌入的学习，从而能够同时考虑色彩分布和像素级详细信息，实现高度真实的图像上色效果，减少颜色错误涂抹的问题，让图片色彩更丰富、自然.
• 特征提取与学习：使用预训练的图像分类模型从输入的黑白图像中提取高级语义特征，包含图像的结构、纹理和对象信息，然后通过深度学习训练，具备自我学习图片内容的能力，能够根据图像本身的特征自动决定合适的颜色填充，而不是依赖于预设规则，极大地提高了图像上色的准确性和自然度.
• 颜色丰富度优化：引入基于颜色平面标准差和均值的损失函数，鼓励生成更加多彩和生动的图像，使上色后的图像色彩更加逼真、丰富

版本特点

• 此为阿里达摩院 CubeAI PyTorch-DDColor图像上色 本地版
• 经测试，上色效果很好，运行速度很快