type
status
date
slug
summary
tags
category
icon
password
DDColor
是由 阿里达摩院 研究的基于深度学习技术的 图像上色模型,它能够自动将黑白或灰度图像着色,使图像更加生动逼真。
ICCV 2023 论文“DDColor:通过双解码器实现逼真的图像着色”的官方 PyTorch 实现
算法整体流程如下图,使用 UNet 结构的骨干网络和图像解码器分别实现图像特征提取和特征图上采样,并利用 Transformer 结构的颜色解码器完成基于视觉语义的颜色查询,最终聚合输出彩色通道预测结果。
该模型适用于多种格式的图像输入,给定黑白图像,生成上色后的彩色图像;给定彩色图像,将自动提取灰度通道作为输入,生成重上色的图像。
- 在线体验
- 官方测试
当然,如果嫌弃自己部署过于麻烦,可以使用下面打包好的程序。
下载地址:https://www.123pan.com/s/7hX0Vv-4Fkbv.html
提取码:ZCYg
- Author:PastKing
- URL:https://www.oaichat.ru/posts/123bbb07-d950-80f3-a7e3-d6f3efe91d7c
- Copyright:All articles in this blog, except for special statements, adopt BY-NC-SA agreement. Please indicate the source!