前言

最近看到一個有意思的機器學習專案——GFPGAN，他可以將模糊的人臉照片恢復清晰，開源專案的Github地址：https://github.com/TencentARC/GFPGAN

我們看一看作者給出的對比圖，

最右側的就是GFPGAN的效果，看一下最左層的輸入圖片，可以發現GFPGAN將圖片恢復的非常清晰，這個效果非常驚艷，

按照以前的慣例，我還是先把這個專案安裝使用一下，看看能不能對代碼重新封裝，變成可以工程化的專案，

環境安裝

我們先看一下專案README給的提示，

首先需要的python版本是>=3.7的，所以我用Anaconda創建了一個python3.9的虛擬環境，Pytorch的安裝直接從官網獲取命令安裝一個最新版本即可，

因為還有一些基礎依賴的安裝，照著安裝一下就行，其實setup.py是已經在專案中的，如下圖，

由于模型比較大，所以作者沒有放在github上，給了下面的下載提示，該模型是作者提供已經訓練好的模型，

如果下載很慢的話，可以從我的網盤下載，

鏈接：https://pan.baidu.com/s/1qU5PifU_qMZfM2Rgv8VpeA
提取碼：TUAN

作者還提供了基礎模型可供自行訓練，

驗證模型

下面我準備了一些圖，挑了一些比較典型的圖片，有黑白的、彩色的以及馬賽克的，想看看是不是都可以實作清晰化處理，

準備的圖片如下：

按照README提供的指令

python inference_gfpgan.py --upscale 2 --test_path inputs/newImages --save_root results

看一下執行結果：

(pytorch39) C:\Users\yi\PycharmProjects\GFPGAN>python inference_gfpgan.py --upscale 2 --test_path inputs/newImages --save_root results
C:\Users\yi\PycharmProjects\GFPGAN\inference_gfpgan.py:45: UserWarning: The unoptimized RealESRGAN is very slow on CPU. We do not use it. If you really want to use it, p
lease modify the corresponding codes.
  warnings.warn('The unoptimized RealESRGAN is very slow on CPU. We do not use it. '
Processing 331.jpg ...
E:\ProgramData\Anaconda3\envs\pytorch39\lib\site-packages\torch\nn\functional.py:3679: UserWarning: The default behavior for interpolate/upsample with float scale_factor
 changed in 1.6.0 to align with other frameworks/libraries, and now uses scale_factor directly, instead of relying on the computed output size. If you wish to restore th
e old behavior, please set recompute_scale_factor=True. See the documentation of nn.Upsample for details.
  warnings.warn(
Processing 333.jpg ...
Processing 334.jpg ...
Processing 335.jpg ...
Results are in the [results] folder.

(pytorch39) C:\Users\yi\PycharmProjects\GFPGAN>

按照默認引數，會在results結果檔案夾中生成4個目錄分別為前后對比圖、原檢測出來的臉部圖、處理后的臉部圖、處理后的最終圖，