微软又一开源力作!专门针对老旧照片

开源
微软研究团队万紫宁、张波等人开发了一种新的基于AI算法,用于通过深度学习恢复老旧照片,与通过监督学习还原旧照片的方式不同,真实照片的降级很复杂。

 微软研究团队万紫宁、张波等人开发了一种新的基于AI算法,用于通过深度学习恢复老旧照片,与通过监督学习还原旧照片的方式不同,真实照片的降级很复杂。

bringing-old-photos-back-to-life作者使用变分自动编码机(VAE)将图像变换到隐空间,并在隐空间进行图像恢复操作。这种转换可以很好的体现出真实照片。

此外,为了解决一张旧照片中的多种退化,bringing-old-photos-back-to-life设计了一个全局分支和局部非全局分支,全局分支专门针对结构化缺陷(例如划痕和灰尘点)的功能,局部非全局分支主要针对非结构化缺陷(例如噪声和模糊度),bringing-old-photos-back-to-life在隐空间中融合了两个分支,从而提高了从多个缺陷还原旧照片的能力。

如何使用

该项目依赖于「Synchronized-BatchNorm-PyTorch」,按照教程配置即可。首先,你需要安装和下载预训练的模型,然后就可以通过一个简单的命令轻松回复旧照片。

对于没有划痕的图像: 

  1. python run.py --input_folder [test_image_folder_path] \  
  2.               --output_folder [output_path] \  
  3.               --GPU 0 

对于有刮擦的图像: 

  1. python run.py --input_folder [test_image_folder_path] \  
  2.               --output_folder [output_path] \  
  3.               --GPU 0 \  
  4.               --with_scratch 

注意:请尝试使用绝对路径。最终结果将保存在中./output_path/final_output/。

目前,微软不打算直接发布带有标签的有划痕旧照片数据集。如果你想要得到配对的数据,你可以使用官方提供的预训练模型来测试收集到的图像,以获得标签。

下面我们还是来看看几组不同照片的还原效果吧:

 

 

责任编辑:庞桂玉 来源: 开源最前线
相关推荐

2020-09-11 15:37:18

GitHub代码开发者

2021-01-26 10:59:52

开源技术 数据

2022-09-30 09:24:27

开源模型

2020-09-26 22:30:18

开源技术 数据

2013-08-01 16:52:31

Tizen移动操作系统移动平台

2021-03-24 10:48:32

谷歌开源代码

2022-08-31 10:11:51

漏洞赏金计划漏洞

2009-04-07 15:31:52

LinuxGoogleMarket

2012-07-19 17:10:03

神奇的阿力Amazing AleRovio

2014-04-28 12:26:54

2015-12-21 13:44:17

2015-10-05 17:38:33

2020-01-20 14:40:39

工具代码开发

2020-11-25 10:34:45

AI 数据工具

2023-04-28 09:18:15

AI技术算法

2020-03-09 09:20:32

开源技术 软件

2014-07-15 09:41:25

Windows 9

2009-03-27 18:24:46

点赞
收藏

51CTO技术栈公众号