照片修复-使用Bringing-Old-Photos-Back-to-Life

2023-10-31 12:40

本文主要是介绍照片修复-使用Bringing-Old-Photos-Back-to-Life,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

项目地址

github项目地址:https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life

环境搭建

1、下载Bringing-Old-Photos-Back-to-Life的github项目
直接下载压缩包解压,或者clone下载都可以。
clone指令:git clone https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life

2、下载工程依赖的其他github项目
依次执行就可以了

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

3、下载预训练模型
人脸识别预训练模型:

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和Global的checkpoints模型:

cd Face_Enhancement/
wget https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Face_Enhancement/checkpoints.zip
unzip checkpoints.zip
cd ../
cd Global/
wget https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Global/checkpoints.zip
unzip checkpoints.zip
cd ../

由于两个checkpoints下载的速度很慢,可以先下载,然后把下载链接复制到迅雷下载,速度可以达到10M。我将下好的上传到了百度云,
地址:https://pan.baidu.com/s/1aZOrp7OF3rt9b8B4UZxTAw,提取码:g1wj
下载完分别放到对应文件夹然后解压。

4、安装环境所需模块
按照github项目下面的说明,直接执行:pip3 install -r requirements.txt就行了。但是问题就来了!cuda、torch、torchvision版本一定要对应,直接pip安装torch和torchvision的话安装的都是最新的,执行修复图片时就会出现下面列举的错误。所以最好的方法就是打开requirements.txt然后将这些包手动pip安装,除了torch和torchvision要指定版本,其他的都可以直接pip安装。

我的环境是:cuda-8.0、torch-1.0.0、torchvision-0.2.2
三者对应关系参考:https://blog.csdn.net/jonado13/article/details/108281688

修复图片

使用时,假设输入路径为 /home/jcy/source/source/old_w_scratch(待修复的图片放在这里),输出路径为/home/jcy/source/source/output_images,那么执行指令为:
图片没有裂痕:

python3 run.py --input_folder /home/jcy/source/source/old_w_scratch --output_
folder /home/jcy/source/source/output_images --GPU 0 

图片有裂痕

python3 run.py --input_folder /home/jcy/source/source/old_w_scratch --output_
folder /home/jcy/source/source/output_images --GPU 0 --with_scratch

执行完毕,修复完的图片就会出现在output_images/final_output中。

注:
修复的图片不要太大,比较大的照片还是别想了,大概率带不动的,大图可以把人身体区域截图下来修复,或者将图片切割或裁剪。

报错
1、incompatible function arguments. The following argument types are supported: 1. (arg0: object, arg1: torch._C._jit_tree_views.SourceRange, arg2: Callable[[str], function]) -> torch._C.Type
错误原因:torch版本和torchvision版本不匹配

2、AssertionError: The NVIDIA driver on your system is too old (found version 9000).
错误原因:torch版本太高,与cuda不匹配,比如我cuda-8.0装了torch-1.5报错了,重新安装torch1.0.0解决。

运行结果

在这里插入图片描述

效果还阔以,但是只生成了这两张,因为这两张图片尺寸比较小,其他找的几张图片比较大,在生成修复图片的时候把我1080ti顶爆了,等啥时候有钱了整个3080ti试试了。。。。硬要修复大的图片,可以裁剪成多份,多次修复以后拼接成一张图片即可,但为了提高拟合度,最好每张图片中都有一块大小适中的重叠区域,而且看了一部分源码,这个项目中似乎对人脸还是挺敏感的,所以最好裁剪的每个部分都把人物包括在里面,以人物为基准向四周拓展。
在这里插入图片描述
修复完一次以后注意清理一下,最好的办法就是把输出图片的文件夹中的所有内容删除(包括图片和文件夹),否则下次修复就会出现重复。

这篇关于照片修复-使用Bringing-Old-Photos-Back-to-Life的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/315421

相关文章

Java 中的 @SneakyThrows 注解使用方法(简化异常处理的利与弊)

《Java中的@SneakyThrows注解使用方法(简化异常处理的利与弊)》为了简化异常处理,Lombok提供了一个强大的注解@SneakyThrows,本文将详细介绍@SneakyThro... 目录1. @SneakyThrows 简介 1.1 什么是 Lombok?2. @SneakyThrows

使用Python和Pyecharts创建交互式地图

《使用Python和Pyecharts创建交互式地图》在数据可视化领域,创建交互式地图是一种强大的方式,可以使受众能够以引人入胜且信息丰富的方式探索地理数据,下面我们看看如何使用Python和Pyec... 目录简介Pyecharts 简介创建上海地图代码说明运行结果总结简介在数据可视化领域,创建交互式地

Java Stream流使用案例深入详解

《JavaStream流使用案例深入详解》:本文主要介绍JavaStream流使用案例详解,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧... 目录前言1. Lambda1.1 语法1.2 没参数只有一条语句或者多条语句1.3 一个参数只有一条语句或者多

Java Spring 中 @PostConstruct 注解使用原理及常见场景

《JavaSpring中@PostConstruct注解使用原理及常见场景》在JavaSpring中,@PostConstruct注解是一个非常实用的功能,它允许开发者在Spring容器完全初... 目录一、@PostConstruct 注解概述二、@PostConstruct 注解的基本使用2.1 基本代

C#使用StackExchange.Redis实现分布式锁的两种方式介绍

《C#使用StackExchange.Redis实现分布式锁的两种方式介绍》分布式锁在集群的架构中发挥着重要的作用,:本文主要介绍C#使用StackExchange.Redis实现分布式锁的... 目录自定义分布式锁获取锁释放锁自动续期StackExchange.Redis分布式锁获取锁释放锁自动续期分布式

springboot使用Scheduling实现动态增删启停定时任务教程

《springboot使用Scheduling实现动态增删启停定时任务教程》:本文主要介绍springboot使用Scheduling实现动态增删启停定时任务教程,具有很好的参考价值,希望对大家有... 目录1、配置定时任务需要的线程池2、创建ScheduledFuture的包装类3、注册定时任务,增加、删

快速修复一个Panic的Linux内核的技巧

《快速修复一个Panic的Linux内核的技巧》Linux系统中运行了不当的mkinitcpio操作导致内核文件不能正常工作,重启的时候,内核启动中止于Panic状态,该怎么解决这个问题呢?下面我们就... 感谢China编程(www.chinasem.cn)网友 鸢一雨音 的投稿写这篇文章是有原因的。为了配置完

使用Python实现矢量路径的压缩、解压与可视化

《使用Python实现矢量路径的压缩、解压与可视化》在图形设计和Web开发中,矢量路径数据的高效存储与传输至关重要,本文将通过一个Python示例,展示如何将复杂的矢量路径命令序列压缩为JSON格式,... 目录引言核心功能概述1. 路径命令解析2. 路径数据压缩3. 路径数据解压4. 可视化代码实现详解1

Pandas透视表(Pivot Table)的具体使用

《Pandas透视表(PivotTable)的具体使用》透视表用于在数据分析和处理过程中进行数据重塑和汇总,本文就来介绍一下Pandas透视表(PivotTable)的具体使用,感兴趣的可以了解一下... 目录前言什么是透视表?使用步骤1. 引入必要的库2. 读取数据3. 创建透视表4. 查看透视表总结前言

Python 交互式可视化的利器Bokeh的使用

《Python交互式可视化的利器Bokeh的使用》Bokeh是一个专注于Web端交互式数据可视化的Python库,本文主要介绍了Python交互式可视化的利器Bokeh的使用,具有一定的参考价值,感... 目录1. Bokeh 简介1.1 为什么选择 Bokeh1.2 安装与环境配置2. Bokeh 基础2