【 B 站 AI 实验室开源】上古老番变4K,超分辨率算法!

机器学习算法与Python实战

共 2156字,需浏览 5分钟

 · 2022-02-17

↓↓↓点击关注,回复资料,10个G的惊喜

机器之心报道 编辑:泽南、小舟

这是 B 站 AI 实验室的最新成果


喜欢看动漫的人经常会被一些经典作品的情节所吸引,不过画面质量是欣赏下去的大敌 —— 那些超过十年历史的内容自不必说,直到现在,很多动画的实际分辨率也只是 720p 左右,然后被拉伸到 1080i 以满足电视台播放的需求,BD(蓝光)清晰度也只有 1080p。

究其原因,还是成本问题:在动画制作过程中,让分辨率提升一个级别,就会导致所有的制作资源大幅提升,从作画、扫描到处理和存储,整个产业链条都需要软硬件的提升。


但今天即使是手机也有个 2K 的分辨率,既然手绘动画难以提升分辨率,那么能不能用 AI 算法来解决这个问题?

最近,GitHub 上一个图像超分辨率的项目火了,一个叫做 Real-CUGAN 的工具可以把动画图像的质量提升 2 到 4 倍,qq 上斗图的表情包也能给你脑补成 4k 品质。


仔细一看,这个项目来自 bilibili 人工智能实验室。上线 20 天,star 量已达 1.5K。


项目链接:https://github.com/bilibili/ailab/tree/main/Real-CUGAN

据作者介绍,Real-CUGAN 是一个使用百万级动漫数据进行训练的,结构与 Waifu2x 兼容的通用动漫图像超分辨率模型。相比目前市面上已有的通用化超分辨率算法,Real-CUGAN 的 AI 模型经过了更大体量数据集的训练,处理二次元内容的效果更佳。

它支持 2x\3x\4x 倍超分辨率,其中 2 倍模型支持 4 种降噪强度与保守修复,3 倍 / 4 倍模型支持 2 种降噪强度与保守修复。

Real-CUGAN 全称为 Real Cascaded-U-Net-style Generative Adversarial Networks(真实的、级联 U-Net 风格的生成对抗网络),使用了与 Waifu2x 相同的动漫网络结构,但因为使用了新的训练数据与训练方法,从而形成了不同的参数。

具体来说,该动漫超分模型训练先行对动漫帧进行切块处理,使用图像质量打分模型对候选块进行打分过滤,得到一个百万级的高质量动漫图像块训练集。然后使用多阶段降质算法,将高清图像块降采样得到低质图像,让 AI 模型学习、优化从低质图像到高质图像的重建过程,训练完毕后即可对真实的二次元低质图像进行高清化处理。

在 GitHub 项目中,作者开源了推理阶段的模型参数与推理代码。

下面,我们来看一下 Real-CUGAN 的修复效果。

图源:https://github.com/bilibili/ailab/tree/main/Real-CUGAN

为了进一步查看几种修复方法的差距,研究团队将 Real-CUGAN 的修复效果和 Waifu2x、RealESRGAN 进行比较。


下图是动漫《侦探已死》中的画面,其中几种方法对沙发纹理的修复差距显著。可以看到,在 Real-CUGAN 的修复结果中,沙发上的横向条纹清晰可见,并且几乎没有中断的部分:


在动漫中,人物清晰的线条对角色形象的塑造具有重要的作用。线条模糊会严重影响观看体验。以下图为例,Real-CUGAN 清晰地修复了动漫中角色的表情线条:


当然,有些场景并非越清晰越好,比如画面中的一些远景,原本就是刻意虚化的。以下图为例,Real-CUGAN 修复之后,近景处的人物形象变得更清晰,远处的蜡烛仍然保持模糊,使得画面更有层次感。


还有一些动漫由于年代久远,受当时技术条件的限制,画质着实「感人」。例如许多人的童年回忆《黑猫警长》,先前的模型 RealESRGAN(下图第三列)已获得了不错的修复效果。相比之下 Real-CUGAN(下图第四列)更好地修复了一些杂线、模糊的问题,让画面更加干净清晰。


更为重要的是这项技术想要用很简单:你下载的模型已训练好,修改 config.py 配置参数,双击 go.bat 运行即可。Real-CUGAN 不仅可用 cuda 加速,也支持 AMD 系列的显卡,处理速度还很快,笔记本电脑也可以跑得起来。

虽然 Real-CUGAN 默认是只处理静态图片的,但很多人已尝试把视频拆解成单帧序列进行处理。如此一来,很多上古老番都能自己上手让它们焕然一新了。

目前 B 站上已经有人上传了一些使用新技术处理过的视频,比如这个在 1080p BD 基础上超分到 4K 的:

在很多电影修复之后,老番高清计划也已在路上。B 站表示,将针对不同品类适配不同的超分算法,并且使用窄带高清算法降低卡顿,提高流畅度。

参考内容:
https://www.bilibili.com/read/cv15031073
  1. 准备写本书
  2. 【虎年大吉】技术年货大礼包
  3. Pytorch 常用损失函数拆解
  4. 吴恩达最新采访:以数据为中心的AI
  5. 人人都能看懂的EM算法推导
  6. 腾讯发布国内首份可解释 AI 报告!

三连在看,月入百万👇

浏览 23
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报