ubuntu20部署Bringing-Old-Photos-Back-to-Life

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ??

环境准备：

ubuntu20.04?

Python 3.8.10

首先将微软的「Bringing-Old-Photos-Back-to-Life」库 clone 到本地：

git clone https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life.git

cd Face_Enhancement/models/networks/
git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm .
cd ../../../

Global 权重配置?

cd Global/detection_models
git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm .
cd ../../

Face_Detection 权重??

人脸识别预训练模型：

cd Face_Detection/
wget http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2
bzip2 -d shape_predictor_68_face_landmarks.dat.bz2
cd ../

Face_Enhancement 权重

Face_Enhancement 权重用于人脸部位增强，下载后权重文件解压至?./Face_Enhancement?

cd Face_Enhancement/
wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/face_checkpoints.zip
unzip face_checkpoints.zip
cd ../
cd Global/
wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/global_checkpoints.zip
unzip global_checkpoints.zip
cd ../

?创建 python3 虚拟环境

python3 -m venv .venv

source .venv/bin/activate

安装项目依赖库

pip3 install -r requirements.txt

出现的问题：

Running Stage 4: Blending
Traceback (most recent call last):
File “align_warp_back_multiple_dlib.py”, line 428, in
blended = blur_blending_cv2(warped_back, blended, backward_mask)
File “align_warp_back_multiple_dlib.py”, line 219, in blur_blending_cv2
mask *= 255.0
numpy.core._exceptions.UFuncTypeError: Cannot cast ufunc ‘multiply’ output from dtype(‘float64’) to dtype(‘uint8’) with casting rule ‘same_kind’

找到文件align_warp_back_multiple_dlib.py的219行，将代码 mask *= 255.0 改为：mask = mask * 255.0，再次运行。
?

修复不带折痕的照片

对于不带折痕的照片，终端输入时需要三个参数，

• --input_folder?，存放需要修复图片的文件路径，不指定时默认为?./test_images/old;
• --output_folder，存放修复后图片的文件路径，不指定时默认为?./output/
• --GPU，指定用到的 GPU 编号，可设定为?0?、?0,1或0,1,2?，没有 GPU 配置时 设为?-1?表示运行时只用 CPU ；?设定 GPU 时需要确保配置的 Pytorch 是 GPU 版本，相对 CPU 的话，GPU 测试时间会更短；

测试时，启动命令如下，这里?--input_folder、--output_folder参数都为默认，所以命令中加这两个参数，电脑比较渣没有 GPU 所以设为 -1；

python3 run.py --GPU -1

?修复带折痕的照片

与不带折痕相比，修复带折痕照片要多加入一个参数?--with_scratch，还有一点不同的是存放原照片默认路径更改为为?./test_images/old_w_scratch?，其它参数设置见 3.1

终端中输入的测试命令如下：

python3 run.py --GPU -1 --with_scratch

最终的运行结果

export PYTHONPATH=/home/xxx/.local/lib/python3.8/site-packages/

文末福利

整个打包版下载地址，联系我要提取码

百度网盘 请输入提取码百度网盘为您提供文件的网络备份、同步和分享服务。空间大、速度快、安全稳固，支持教育网加速，支持手机端。注册使用百度网盘即可享受免费存储空间https://pan.baidu.com/s/1oKgBB0gkRIGjwzeTtFYWFw百度网盘 请输入提取码