DDColor 是一款开源的黑白图像上色模型,输入一张黑白图像,实现端到端的全图上色,返回上色处理后的彩色图像。DDColor 是最新的 SOTA 图像上色算法,能够对输入的黑白图像生成自然生动的彩色结果。算法整体流程如下图,使用 UNet 结构的骨干网络和图像解码器分别实现图像特征提取和特征图上采样,并利用 Transformer 结构的颜色解码器完成基于视觉语义的颜色查询,最终聚合输出彩色通道预测结果。
[t-success icon='']DDColor[/t-success]
DDColor目前已经公开了论文和代码,此模型只需要上传一张黑白图片,就能还你一张彩色图片。非常适合为老照片添加颜色,开发者已经在ModelScope(魔塔社区) 上线了Demo,即开即用。
GitHub地址:https://github.com/piddnad/DDColor
论文地址:https://arxiv.org/abs/2212.11613
ModelScope地址:https://modelscope.cn/models/damo/cv_ddcolor_image-colorization/summary
如何使用
在 ModelScope 框架上,提供输入图片,即可以通过简单的 Pipeline 调用来使用图像上色模型。