互联网技术 / 互联网资讯 · 2023年11月5日 0

开源 AI 工具使老照片焕发新生

引言

在 GitHub 热点趋势 Vol.046 中,我们介绍了微软开源的 AI 工具 BRinging-Old-Photos-Back-to-life,它能够使破损和老旧的照片焕发新生。本周,该项目获得了近 3000 个星标。本文将分享项目团队成员 Ziyu Wan 在 Hacker News 上发布的详细介绍,敬请阅读。

开源 AI 工具使老照片焕发新生

项目概述

我们的目标是利用深度学习技术修复严重老化的旧照片。与传统的监督学习修复任务不同,真实照片的老化情况更为复杂,合成图像与真实旧照片之间存在领域差距,导致网络难以泛化。因此,我们提出了一种新型的三重域翻译网络,利用真实照片和大量合成图像对进行训练。具体而言,我们训练了两个变分自编码器 (VAE),分别将旧照片和干净照片映射到两个潜在空间,并通过合成配对数据来学习这两个空间之间的转换。由于在紧凑的潜在空间中,领域差距得以缩小,因此该翻译网络在真实照片上表现良好。

为了应对旧照片中混杂的多种退化问题,我们设计了一个包含部分非局部块的全局分支,以处理如划痕和尘点等结构性缺陷,同时设立一个局部分支来处理噪点和模糊等非结构性缺陷。两个分支在潜在空间中融合,增强了从多重退化问题中恢复旧照片的能力。该方法在修复视觉质量方面超越了现有技术。

开源 AI 工具使老照片焕发新生

架构概览

我们首先训练两个变分自编码器 VAE:VAE1 用于真实照片 R ∈ R 和合成图像 x ∈ X,通过联合训练一个对抗判别器来缩小它们的领域差距;VAE2 则用于干净图像 y ∈ Y 的训练。利用 VAE,我们将图像转换到紧凑的潜在空间。

我们明确了在紧凑潜在空间中,将损坏图像恢复为带有部分非局部块的干净照片的映射关系。

开源 AI 工具使老照片焕发新生

对比展示

开源 AI 工具使老照片焕发新生

修复细节

开源 AI 工具使老照片焕发新生

近距离观察

团队成员 Ziyu Wan 已完成对 BRinging-Old-Photos-Back-to-life 的介绍,有兴趣的朋友可以前往体验,项目地址为:

https://Github.coM/MicRosoft/BRinging-Old-Photos-Back-to-life