Python玩家使用B站开源项目Real-CUGAN修复动漫图片、视频

发现B站开源了一个可以修复二次元动漫图片、视频的项目,话不多说,看对比图:
原图(600x600):
在这里插入图片描述
2倍超分图(1200x1200):
在这里插入图片描述
4倍超分图(2400x2400):
在这里插入图片描述


这是提供给有Python基础的人玩的,能玩到什么程度就看各位大佬了。

环境搭建

Real-CUGAN 为Windows用户打包了一个可执行环境。
Windows玩家可参考此博客:【图片、视频修复】Real-CUGAN的简单使用

代码下载

Github项目地址:https://github.com/bilibili/ailab
进入链接地址后可使用如下方法下载源码
在这里插入图片描述注:国内直接访问Github不稳定,可使用代理解决

下载或解压完成后可得到如下文件,Real-CUGAN就是项目文件夹了。
在这里插入图片描述
进入项目文件后如图所示(自己玩用这几个文件就够了,大神可以去试着用一下VaporuSynth文件内的东西):
在这里插入图片描述

模型下载

以下是官方提供的超分工具下载链接
百度网盘(提取码ds2a) 🔗GithubRelease 🔗 | 和彩云(提取码tEr1,手机号验证码登录,不限速无需客户端) 🔗GoogleDrive 🔗

我使用的是和彩云下载在这里插入图片描述下载完成后,将解压文件放入项目文件内的模型文件夹内:
在这里插入图片描述

PyTorch安装

安装PyTorch时我参考了这篇文章:
conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

为了保证PyTorch能成功调用显卡CUDA,需要
1、确定显卡是否支持CUDA(如何查看显卡是否支持CUDA,以及查看CUDA的版本?
2、确保驱动升级到最新(确保驱动与cuda版本互相支持)

CUDA安装包挺大的,安装的也很慢。

这里用的显卡是NVIDIA GeForce GTX 1050 Ti
CUDA版本是 11.6
在这里插入图片描述

验证CUDA可用:

win+R输入cmd进入控制窗同,输入 nvcc -V ,出现如图所示代表成功:
在这里插入图片描述

我在手动指定PyTorch版本的时候一直出现无法调用的情况,最终使用PyTorch官方建议的命令才得以解决:
PyTorch官网:https://pytorch.org/

官网提供了多种下载方案,根据自己的需求下载即可(官网还没有CUDA 11.6的版本,所以我选择了最新的11.3版本):
在这里插入图片描述在这里插入图片描述
torch的文件非常大,最大的有1.7G,建议使用虚拟环境安装

验证PyTorch可用:

import torch
print(torch.cuda.is_available())

# 若返回True,则证明CUDA可用

在这里插入图片描述
以上的配制都是容易出错的,不同的电脑会有所不同,像是torch与CUDA版本不兼容会一直False,当时为了解决这个问题花了不少时间。。。

安装环境依赖

下图是官方给出的环境依赖,torch也就是上面安装的PyTorch,不需要再次安装。
在这里插入图片描述

可以根据自己的习惯安装Python库,我习惯用pip直接安装:

pip install numpy opencv-python moviepy

环境的搭建到此完成,想要在自己电脑上正常运行还需要进行调试,可以通过运行upcunet_v3.py超分图片(或运行inference_video.py超分视频),根据报错信息修改代码就行了。

报错

找不到指定的模块

在这里插入图片描述解决方案:Importing the numpy c-extensions failed 解决方案

  • 5
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

胡桃の壶

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值