【老照片上色+在线运行】DDColor:阿里黑白图像上色模型(2022.12提出)

[2024-01-18] 在线运行demo: https://www.modelscope.cn/models/iic/cv_ddcolor_image-colorization/summary

[2023-12-13] Release the DDColor-tiny pre-trained model!

[2023-09-07] Model Zoo 多个模型发布:https://github.com/piddnad/DDColor/blob/master/MODEL_ZOO.md

[2023-05-15] 代码开源https://github.com/piddnad/DDColor?tab=readme-ov-file

[2022-12-22] 发布论文 DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

简介

阿里巴巴摩院 提出基于Unet+Transfomer(cross-attenrtion)。双解码结构的图像上色着色image colorization)模型

由于图片残缺脱色破损等病态,多模态的信息的不确定性,自动图像着色是一个具有挑战性的问题。

直接训练深度神经网络通常会导致语义颜色不正确,颜色丰富度低(low color richness)

不像以前的方法是通过额外的网络或手动计算先验来优化颜色可能性,我们的方法使用基于变换的颜色解码器以端到端方式学习自适应语义感知的颜色嵌入。通过使用多尺度图像特征来学习颜色查询,我们的方法减轻了颜色溢出,并显著改善了小物体的着色(见图1)。在此基础上,我们提出了一种新的色彩损失,以提高生成结果的色彩丰富度。

主要方法

为了使灰度图像XL上色,我们的模型首先使用骨干网络(ConvNeXt )提取其视觉特征。然后将这些特征输入图像解码器,解码器恢复图像的空间结构。

同时,颜色解码器利用图像解码器产生的不同尺度的图像特征学习自适应颜色查询(adaptive color queries)。
融合模块(fusion)将两个解码器产生的图像和颜色特征结合起来,产生色彩生动,语义清晰 (vivid and semantic-aware color)的颜色输出。

最后,我们沿着通道维度(channel dimension)将y^AB和xL 串联起来,得到最终的着色结果。

(b)彩色解码器块的结构。以图像特征和**可学习**的颜色查询为输入,通过交叉注意、自注意和前馈操作(feed forward operations)建立语义和颜色表示之间的相关性

具体的:包括一个多尺度(multi-scale) 图像解码器和一个基于transformer的颜色解码器( color decoder)。

前者恢复图像的空间分辨率,而后者通过交叉注意(cross-attention)建立颜色和语义表示之间的相关性。

我们的两个解码器不是使用额外的先验,而是协同工作以利用多尺度图像特征来指导自适应颜色查询的优化,从而显著减轻颜色溢出(color bleeding effects)效果。此外,还引入了一种简单而有效的色彩损失(colorfulness loss),以进一步提高生成结果的颜色丰富度。

实测

国内阿里demo(打开界面 右侧比较小):

https://www.modelscope.cn/models/iic/cv_ddcolor_image-colorization/summary

replicate

https://replicate.com/piddnad/ddcolor

本机部署测试

相关推荐
阿_旭14 天前
一文读懂| 自注意力与交叉注意力机制在计算机视觉中作用与基本原理
人工智能·深度学习·计算机视觉·cross-attention·self-attention
老艾的AI世界2 个月前
AI老照片修复神器,Anole下载介绍
图像处理·人工智能·深度学习·神经网络·目标检测·机器学习·ai·老照片上色·老照片修复·ai老照片修复
凉_白开7 个月前
阿里开源黑白图片上色算法DDColor的部署与测试并将模型转onnx后用c++推理
c++·onnx·ddcolor
_长银1 年前
GitHub项目推荐-Deoldify
ai作画·github·老照片上色