AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

“视频闪烁”(比如画面突然一白)有新招了

BlazeBVD团队 投稿

量子位 | 公众号 QbitAI

家人们,消除“视频闪烁”(比如画面突然一白)有新招了!

回想一下,当你看一部老电影或者用手机拍摄的视频时,画面偶尔会出现闪烁或颜色不一致等现象。

为了消除这些,来自美图影像研究院、中国科学院大学,以及四川大学的研究人员提出了一种新算法

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

新算法“BlazeBVD”可以自动消除视频中的闪烁,而且处理速度非常快,据称比现有方法快10倍

更妙的是!这种方法甚至无需事先知道视频闪烁的具体类型或程度。

换句话说,它是“盲”的,可以应用于各种不同的视频。

这下,即使拍摄环境光线变化,或者相机硬件跟不上也无需头疼了。[doge]

目前相关论文已被计算机视觉顶会ECCV 2024接收。

感兴趣的话,咱们接着康康~

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

BlazeBVD如何消除视频闪烁?

首先,受经典的STE(闪烁去除方法尺度时间均衡)启发,BlazeBVD引入了直方图辅助解决方案。

图像直方图被定义为像素值的分布,它被广泛应用于图像处理,以调整图像的亮度或对比度。

打个比方,图像直方图就像是一个统计表,它告诉我们在一张照片中,不同亮度的像素有多少。

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

△ 图片由Claude 3.5 Sonnet生成

而STE通过分析视频中每一帧的直方图,然后用一种叫做高斯滤波的方法来平滑这些直方图,先初步矫正直方图分布突变的图像帧,能够让画面看起来更加稳定,减少闪烁。

虽然STE只对一些轻微的闪烁有效,但它验证了

  • 直方图比原始的像素数据更简洁,能够更有效地捕捉到视频中的亮度变化和闪烁。
  • 通过平滑直方图,可以减少视频中的闪烁现象,让视频看起来更稳定。

因此,利用STE和直方图的提示来提高盲视频去闪烁的质量和速度是可行的。

具体而言,BlazeBVD包括三个阶段。

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

BlazeBVD三阶段详解

就像医生治病一样,BlazeBVD会首先检查视频的每一帧。

它引入了STE对视频帧在光照空间下的直方图序列进行校正。

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

然后从处理过的帧中提取出重要信息,比如哪些帧闪烁最明显(奇异帧集)、哪些地方光线需要调整(滤波后的光照图),以及哪些地方曝光过度或不足(曝光图)

接下来,BlazeBVD开始进行修复

一方面,BlazeBVD使用一个叫做全局闪烁去除模块(GFRM)的工具,利用之前提取的光照图来调整整个视频的光线,确保每一帧的亮度和颜色看起来都很自然。

另一方面,对于一些特别需要关注的局部区域,比如曝光过度或不足的地方,BlazeBVD会使用局部闪烁去除模块(LFRM)。这个模块会利用光流信息(就像追踪物体在视频中的运动)来修复这些区域的细节。

完成这一步,BlazeBVD最后进行完善工作

它引入一个轻量级的时序网络(TCM),这个网络就像是视频的“美容师”,确保每一帧在视觉上都是平滑过渡的,没有突兀的变化。

为了进一步提高视频的一致性,BlazeBVD设计了一种特殊的评分系统(自适应掩模加权损失)。这个系统会给每一帧打分,确保它们在视觉上的一致性,让整个视频看起来更加流畅和自然。

至此,BlazeBVD完成了整个“诊治”流程。

实验结果

那么,BlazeBVD究竟效果如何呢?

直接看已有方法与BlazeBVD在盲视频去闪烁任务上的结果对比:

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

其中Deflicker为已有方法,GT(Ground Truth)表示理想的无闪烁视频,而KL散度表示处理后的视频与理想无闪烁视频之间的差异。KL数值越大,差异越大。

可以看出,BlazeBVD能很好地恢复照明直方图,同时避免出现颜色伪影和颜色失真(例如第二列男人的手臂)

再进一步和基线方法进行量化对比:

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

BlazeBVD在PSNR(峰值信噪比,数值越高表示视频质量越好)和SSIM(结构相似性指数,数值接近1表示视频质量越好)上得分较高,且在Ewarp(数值越低,视频越连贯一致)得分较低。

一句话,BlazeBVD超越了已有基线方法。

为了直观展现这种差别,BlazeBVD与基线方法的可视化对比如下:

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

消融实验也验证了BlazeBVD所设计模块的有效性:

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

概括而言,通过对合成视频、真实视频和生成视频的综合实验,BlazeBVD展现了优越的定性和定量结果,并且比最先进的模型推理速度快10倍。

AI视频修复速度10倍提升,过曝变色也能逐帧搞定|美图国科大新算法

目前相关论文已公开,感兴趣可以进一步了解。

论文:
https://arxiv.org/html/2403.06243v1

版权所有,未经授权不得以任何形式转载及使用,违者必究。