關(guān)注上方“GitHuboy”，選擇星標(biāo)，

關(guān)鍵時(shí)間，第一時(shí)間送達(dá)！

大家好，我是boy哥。

最近微軟又開源一個(gè)力作「Bringing-Old-Photos-Back-to-Life」，可以修復(fù)舊照片，讓舊照片重現(xiàn)原貌。

這個(gè)開源項(xiàng)已發(fā)表論文，分別在Cornell大學(xué)的CVPR和雜志上刊登。

▍Demo測試

GitHub開源項(xiàng)目上已提供了一個(gè) Colab Demo，感興趣的可在這個(gè) Demo 上使用自己的舊照片嘗試恢復(fù)一下。

Colab Demo 地址：https://colab.research.google.com/drive/1NEm6AsybIiC5TwTU_4DqDkQO0nFRB-uA?usp=sharing

下面是是一些被成功修復(fù)的老圖片，可以自己對(duì)比下新老照片效果。

比如原來有裂紋的照片可以完美修復(fù)掉，顏色上會(huì)變得更加鮮明，完全看不出舊的痕跡了。

▍要求

該代碼已在裝有Nvidia GPU和CUDA的Ubuntu上進(jìn)行了測試。需要Python> = 3.6才能運(yùn)行代碼。

▍如何使用

1、 Full Pipeline

在安裝和下載預(yù)訓(xùn)練的模型后，你可以使用一個(gè)簡單的命令輕松恢復(fù)舊照片。

對(duì)于沒有劃痕的圖像：

python?run.py?--input_folder?[test_image_folder_path]?\
?????????????????????????--output_folder?[output_path]?\
?????????????????????????--GPU?0

對(duì)于有劃痕的圖像：

python?run.py?--input_folder?[test_image_folder_path]?\
?????????????????????????--output_folder?[output_path]?\
?????????????????????????--GPU?0?\
?????????????????????????--with_scratch

2、Scratch Detection

目前，我們不打算直接發(fā)布帶有標(biāo)簽的有劃痕舊照片數(shù)據(jù)集。如果你想要得到配對(duì)的數(shù)據(jù)，你可以使用我們的預(yù)訓(xùn)練模型來測試收集到的圖像，以獲得標(biāo)簽。

cd?Global/
python?detection.py?--test_path?[test_image_folder_path]?\
????????????????????????????????????--output_dir?[output_path]?\
????????????????????????????????????--input_size?[resize_256|full_size|scale_256]

3、Global Restoration

提出了三重態(tài)域翻譯網(wǎng)絡(luò)來解決舊照片的結(jié)構(gòu)化退化和非結(jié)構(gòu)化退化。

cd?Global/
python?test.py?--Scratch_and_Quality_restore?\
???????????????--test_input?[test_image_folder_path]?\
???????????????--test_mask?[corresponding?mask]?\
???????????????--outputs_dir?[output_path]

python?test.py?--Quality_restore?\
?--test_input?[test_image_folder_path]?\
?--outputs_dir?[output_path]