DeOldify图像上色

我要开发同款
匿名用户2024年07月31日
148阅读

技术信息

开源地址
https://modelscope.cn/models/iic/cv_unet_image-colorization
授权协议
Apache License 2.0

作品详情

图像上色介绍

输入一张黑白图片,全自动的输出相对应的彩色图片。

模型描述

DeOldify是图像上色领域比较有名的开源算法,模型利用reset作为ecoder构建一个uet结构的网络,并提出了多个不同的训练版本,在效果、效率、鲁棒性等等方面有良好的综合表现。

模型结构

期望模型使用方式以及适用范围

本模型适用范围较广,给定任意的图片,都能生成一张上色后的彩色图片。如果输入的是彩色图片,将进行重上色。

如何使用

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

代码范例

import cv2
from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks

colorizer = pipelie(Tasks.image_colorizatio, model='damo/cv_uet_image-colorizatio')
result = colorizer('https://modelscope.oss-c-beijig.aliyucs.com/test/images/marily_moroe_4.jpg')
cv2.imwrite('result.pg', result['output_img'])

模型局限性以及可能的偏差

  • 本算法使用的是稳定版本的DeOldify模型,在色彩鲜艳度上可能会稍逊一些。
  • 因训练数据、模型结构等限制,算法存在一些生成瑕疵效果。

数据评估及结果

本算法主要在ImageNetCOCO-StuffCelebA-HQ上测试。

Metric ImageNet COCO-Stuff CelebA-HQ
FID 3.87 13.86 9.48
PSNR 22.97 24.19 25.20

相关论文以及引用信息

@misc{deoldify,
    author    = {J. Atic},
    title     = {A deep learig based project for colorizig ad restorig old images (ad video!)},
    howpublished = "\url{https://github.com/jatic/DeOldify}",
}

功能介绍

图像上色介绍 输入一张黑白图片,全自动的输出相对应的彩色图片。 模型描述 DeOldify是图像上色领域比较有名的开源算法,模型利用resnet作为encoder构建一个unet结构的网络,并提出了多

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论