pytorch框架实现老照片修复功能详细演示(GPU版)

一、环境说明

环境	版本
系统	windows10
显卡	RTX3060
CUDA	11.1.1
cuDNN	8.1.1
Python	3.8.8
torch	1.9.0+cu111
torchvision	0.10.0+cu111
torchaudio	0.9.0

cuda安装参考： tensorflow-gpu2.4.1安装配置详细步骤

算法模型下载地址：链接：https://pan.baidu.com/s/1uGNXd5g2dKjb4zgckxiLdA 
提取码：vqv6

二、安装依赖模块

1.创建新的虚拟环境。

2.安装torch、torchvision、torchaudio模块。

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

3.安装依赖库。

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

4.在./Bringing-Old-Photos-Back-to-Life/input文件夹中存放需要修复的照片

 并执行如下命令：

python run.py --input_folder C:\Users\Administrator\Desktop\Bringing-Old-Photos-Back-to-Life\input --output_folder C:\Users\Administrator\Desktop\Bringing-Old-Photos-Back-to-Life\output --GPU 0

执行结果：

 生成修复后的照片路径：

 效果对比：

修复前照片

修复后照片