NTU S-Lab 开源的图像修复/增强 AI软件

新加坡南洋理工大学S-Lab实验室在GitHub上开源了一份图像修复和增强的AI软件。

链接如下：

GitHub - sczhou/CodeFormer: [NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

下面简单介绍如何使用。

1.首先下载代码到本地。

2. 下载和安装依赖。

代码附带的basicsr库也要手动安装。

3. 安装好之后可以在命令行来调用软件进行图像修复和增强。

3.1 单张人脸修复

python inference_codeformer.py -w 0.5 --has_aligned --input_path空格加人脸图片文件夹路径，例如

python inference_codeformer.py -w 0.5 --has_aligned --input_path K:\CodeFormer-master\MyInputs\image

效果如下：

3.2 整体图像增强

python inference_codeformer.py -w 0.7 --input_path空格加图片文件夹路径，例如：

python inference_codeformer.py -w 0.7 --input_path K:\CodeFormer-master\MyInputs\image

支持的图片格式：'jpg', 'png'

效果如下：

张自忠将军

3.3 视频增强

python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path空格加视频文件夹路径

支持的视频格式: 'mp4', 'mov', 'avi'

暂时还没找到好的模糊视频，这里不再举例。

软件不足的地方是目前不支持中文路径。