【老照片上色+在线运行】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)建立语义和颜色表示之间的相关性
【老照片上色+在线运行】DDColor:阿里黑白图像上色模型(2022.12提出)_第1张图片
具体的:包括一个多尺度(multi-scale) 图像解码器和一个基于transformer的颜色解码器( color decoder)。
前者恢复图像的空间分辨率,而后者通过交叉注意(cross-attention)建立颜色和语义表示之间的相关性。
我们的两个解码器不是使用额外的先验,而是协同工作以利用多尺度图像特征来指导自适应颜色查询的优化,从而显著减轻颜色溢出(color bleeding effects)效果。此外,还引入了一种简单而有效的色彩损失(colorfulness loss),以进一步提高生成结果的颜色丰富度。

实测

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

https://www.modelscope.cn/models/iic/cv_ddcolor_image-colorization/summary
【老照片上色+在线运行】DDColor:阿里黑白图像上色模型(2022.12提出)_第2张图片
【老照片上色+在线运行】DDColor:阿里黑白图像上色模型(2022.12提出)_第3张图片

replicate

https://replicate.com/piddnad/ddcolor

本机部署测试

你可能感兴趣的:(AIGC,生成式人工智能,论文解读,老照片上色,colorization,阿里达摩院,DDColor,cross-attention)