纯净、安全、绿色的下载网站

首页|软件分类|下载排行|最新软件|IT学院

当前位置:首页IT学院IT技术

Python照片修复 Python 人工智能老照片修复算法学习

剑客阿良_ALiang   2021-11-18 我要评论
想了解Python 人工智能老照片修复算法学习的相关内容吗剑客阿良_ALiang在本文为您仔细讲解Python照片修复的相关知识和一些Code实例欢迎阅读和指正我们先划重点:Python,人工智能,Python照片修复下面大家一起来学习吧。

前言

老旧或者破损的照片如何修复呢?本文主要分享一个博主使用后非常不错的照片恢复开源项目:Bringing-Old-Photos-Back-to-Life。

项目的Github地址:项目地址

我们先看看官方给出的效果图:

就算现在看到这张图我仍然觉着非常惊艳。下面我会把项目环境安装部署到最后使用的效果做一个展示。

项目环境搭建

该项目的环境搭建有点复杂我一点点说。

conda虚拟环境创建

在项目README.md文件中要求python版本在3.6以上。

我们用anaconda创建一个虚拟环境bobl

conda create -n bobl python=3.6

 激活环境

conda activate bobl

在Pycharm项目中配置interpreter设置到conda目录envs下bobl环境的python。

Pytorch安装

虽然项目官方给出的requirements.txt包含pytorch为了保险起见还是去Pytorch官方网站上安装一下。Pytorch官方地址:PyTorch

因为我本机没插显卡安装的cpu版本。

选择对应的命令安装Pytorch库。

Synchronized-BatchNorm-PyTorch repository安装

 官方给出的安装说明里面需要部署Synchronized-BatchNorm-PyTorch项目进来。

这里注意一点需要把Synchronized-BatchNorm-PyTorch项目中的sync_batchnorm拷贝到上级目录。完整的目录接口参考下图:

Global目录Synchronized-BatchNorm-PyTorch项目部署

官方说明里面也需要把Synchronized-BatchNorm-PyTorch项目部署到Global里面。

也是一样要把sync_batchnorm拷贝到上级目录。结构如下:

检测预处理模型下载

需要用到一个检测预处理模型主要是用来识别照片中的人脸部分的。

 注意解压后的位置结构如下:

下载脸部增强模型文件

官方说明:

下载两个模型zip解压到对应目录下结构如下:

下载依赖

注意我这里去掉了pytorch的依赖安装已经已经装过了。

dlib
scikit-image
easydict
PyYAML
dominate>=2.3.1
dill
tensorboardX
scipy
opencv-python
einops
PySimpleGUI

安装命令:

pip install -r requirements.txt -i https://pypi.douban.com/simple

完整部署后项目结构

完整的结构如下图:

项目使用

官方给到的图我就不用了不能说明问题。我自己准备了两种图一张是一张人脸的一张是多张人脸的。

 

先按照官方给出的命令跑跑看

我们直接使用最下面这个命令包含划痕去除与高度还原。看一下执行情况。

(bobl) D:\spyder\Bringing-Old-Photos-Back-to-Life>python run.py --input_folder E:\csdn\老照片 --output_folder result1/ --GPU -1 --with_scratch --HR
Running Stage 1: Overall restoration
initializing the dataloader
model weights loaded
directory of testing image: E:\csdn\老照片
processing 1.jpg
processing 2.jpg
Mapping: You are using multi-scale patch attention, conv combine + mask input
Now you are processing 1.png
C:\ProgramData\Anaconda3\envs\bobl\lib\site-packages\torch\nn\functional.py:3635: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4
.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
  "See the documentation of nn.Upsample for details.".format(mode)
Now you are processing 2.png
Finish Stage 1 ...
 
 
Running Stage 2: Face Detection
12
1
Finish Stage 2 ...
 
 
Running Stage 3: Face Enhancement
dataset [FaceTestDataset] of size 13 was created
The size of the latent vector size is [16,16]
Network [SPADEGenerator] was created. Total number of parameters: 92.1 million. To see the architecture, do print(network).
hi :)
Finish Stage 3 ...
 
 
Running Stage 4: Blending
Finish Stage 4 ...
 
 
All the processing is done. Please check the results.
 
(bobl) D:\spyder\Bringing-Old-Photos-Back-to-Life>

输出的文件不但有最终的结果也有检测出来的每个脸的处理前后效果。结果结构如下:

验证一下

1、多人照片最终的效果验证下面上图是输出结果图下图是原始图。可以看出有些划痕已经消失但是还是有一些不过整体的任务更立体鲜明了。

2、单人照片最终效果验证下面上图为结果图下图为原始图。单人就很明显了不但划痕都消除了人也更清晰立体效果是真不错。

3、模型也会把多人图中的每张脸都识别出来并且跑出结果可以对比一下看看细节还是修复的很好的。

 

总结

官方还给出了其他的命令就不一一验证了。整体的效果是非常好的只是在多人图的时候还有些瑕疵瑕不掩瑜。


相关文章

猜您喜欢

  • C#圆角按钮 C#实现自定义圆角按钮的方法

    想了解C#实现自定义圆角按钮的方法的相关内容吗胖达没有海在本文为您仔细讲解C#圆角按钮的相关知识和一些Code实例欢迎阅读和指正我们先划重点:C#圆角按钮,C#自定义圆角按钮下面大家一起来学习吧。..
  • Android Compose底部按钮 Android Compose实现底部按钮以及首页内容详细过程

    想了解Android Compose实现底部按钮以及首页内容详细过程的相关内容吗theyangchoi在本文为您仔细讲解Android Compose底部按钮的相关知识和一些Code实例欢迎阅读和指正我们先划重点:Android,底部按钮,Android,Compose底部按钮下面大家一起来学习吧。..

网友评论

Copyright 2020 www.tdogsoftware.com 【零度软件园】 版权所有 软件发布

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 点此查看联系方式