當(dāng)前位置：首頁 > 编程语言 > python >内容正文

python

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

發(fā)布時(shí)間：2023/12/20 python 29 豆豆

生活随笔收集整理的這篇文章主要介紹了 Python玩家使用B站开源项目Real-CUGAN修复动漫图片、视频小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.

發(fā)現(xiàn)B站開源了一個可以修復(fù)二次元動漫圖片、視頻的項(xiàng)目，話不多說，看對比圖：
原圖（600x600）：

2倍超分圖（1200x1200）：

4倍超分圖（2400x2400）：

這是提供給有Python基礎(chǔ)的人玩的，能玩到什么程度就看各位大佬了。

Real-CUGAN 為Windows用戶打包了一個可執(zhí)行環(huán)境。
Windows玩家可參考此博客：【圖片、視頻修復(fù)】Real-CUGAN的簡單使用

Github項(xiàng)目地址：https://github.com/bilibili/ailab
進(jìn)入鏈接地址后可使用如下方法下載源碼
注：國內(nèi)直接訪問Github不穩(wěn)定，可使用代理解決

下載或解壓完成后可得到如下文件，Real-CUGAN就是項(xiàng)目文件夾了。

進(jìn)入項(xiàng)目文件后如圖所示（自己玩用這幾個文件就夠了，大神可以去試著用一下VaporuSynth文件內(nèi)的東西）：

以下是官方提供的超分工具下載鏈接
百度網(wǎng)盤(提取碼ds2a) 🔗｜GithubRelease 🔗 | 和彩云(提取碼tEr1,手機(jī)號驗(yàn)證碼登錄,不限速無需客戶端) 🔗｜ GoogleDrive 🔗

我使用的是和彩云下載下載完成后，將解壓文件放入項(xiàng)目文件內(nèi)的模型文件夾內(nèi)：

安裝PyTorch時(shí)我參考了這篇文章：
conda下安裝pytorch最詳細(xì)教程 // 安裝pytorch踩坑記錄 // cuda11.5下pytorch安裝 // torch.cuda.is_available()輸出False解決辦法

為了保證PyTorch能成功調(diào)用顯卡CUDA，需要
1、確定顯卡是否支持CUDA（如何查看顯卡是否支持CUDA，以及查看CUDA的版本？）
2、確保驅(qū)動升級到最新（確保驅(qū)動與cuda版本互相支持）

CUDA安裝包挺大的，安裝的也很慢。

這里用的顯卡是NVIDIA GeForce GTX 1050 Ti
CUDA版本是 11.6

win+R輸入cmd進(jìn)入控制窗同，輸入 nvcc -V ，出現(xiàn)如圖所示代表成功：

我在手動指定PyTorch版本的時(shí)候一直出現(xiàn)無法調(diào)用的情況，最終使用PyTorch官方建議的命令才得以解決：
PyTorch官網(wǎng)：https://pytorch.org/

官網(wǎng)提供了多種下載方案，根據(jù)自己的需求下載即可（官網(wǎng)還沒有CUDA 11.6的版本，所以我選擇了最新的11.3版本）：

torch的文件非常大，最大的有1.7G，建議使用虛擬環(huán)境安裝

import torch print(torch.cuda.is_available())# 若返回True，則證明CUDA可用

以上的配制都是容易出錯的，不同的電腦會有所不同，像是torch與CUDA版本不兼容會一直False，當(dāng)時(shí)為了解決這個問題花了不少時(shí)間。。。

下圖是官方給出的環(huán)境依賴，torch也就是上面安裝的PyTorch，不需要再次安裝。

可以根據(jù)自己的習(xí)慣安裝Python庫，我習(xí)慣用pip直接安裝：

pip install numpy opencv-python moviepy

環(huán)境的搭建到此完成，想要在自己電腦上正常運(yùn)行還需要進(jìn)行調(diào)試，可以通過運(yùn)行upcunet_v3.py超分圖片（或運(yùn)行inference_video.py超分視頻），根據(jù)報(bào)錯信息修改代碼就行了。

解決方案：Importing the numpy c-extensions failed 解決方案

以上是生活随笔為你收集整理的Python玩家使用B站开源项目Real-CUGAN修复动漫图片、视频的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。