Python玩家使用B站开源项目Real-CUGAN修复动漫图片、视频
發現B站開源了一個可以修復二次元動漫圖片、視頻的項目,話不多說,看對比圖:
原圖(600x600):
2倍超分圖(1200x1200):
4倍超分圖(2400x2400):
B站開源項目Real-CUGAN
- 環境搭建
- 代碼下載
- 模型下載
- PyTorch安裝
- 驗證CUDA可用:
- 驗證PyTorch可用:
- 安裝環境依賴
- 報錯
- 找不到指定的模塊
這是提供給有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 解決方案
總結
以上是生活随笔為你收集整理的Python玩家使用B站开源项目Real-CUGAN修复动漫图片、视频的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 华为HCNE考试练习题经典模拟题
- 下一篇: react根据浏览器的尺寸动态调整布局