照片级、真实感的黑白图像上色模型DDColor,为老照片添加生动、自然的色彩

345次阅读
没有评论

共计 598 个字符,预计需要花费 2 分钟才能阅读完成。

DDColor 是一款开源的黑白图像上色模型,输入一张黑白图像,实现端到端的全图上色,返回上色处理后的彩色图像。DDColor 是最新的 SOTA 图像上色算法,能够对输入的黑白图像生成自然生动的彩色结果。算法整体流程如下图,使用 UNet 结构的骨干网络和图像解码器分别实现图像特征提取和特征图上采样,并利用 Transformer 结构的颜色解码器完成基于视觉语义的颜色查询,最终聚合输出彩色通道预测结果。

照片级、真实感的黑白图像上色模型 DDColor,为老照片添加生动、自然的色彩

DDColor

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

Colab 在线体验:https://github.com/camenduru/DDColor-colab

如何使用

在 ModelScope 框架上,提供输入图片,即可以通过简单的 Pipeline 调用来使用图像上色模型。

照片级、真实感的黑白图像上色模型 DDColor,为老照片添加生动、自然的色彩

正文完
关注公众号获取最新教程
post-qrcode
 
AI小诸葛
版权声明:本站原创文章,由 AI小诸葛 2024-01-17发表,共计598字。
转载说明:除特殊说明外本站文章皆由原创发布,请勿转载。