Vapoursynth視頻壓制教程 系列二:Real cugan動(dòng)漫超分辨率視頻的壓制之pytorch篇
這一期主要講Vapoursynth和real-cugan以及ffmpeg壓制流程。Vapoursynth運(yùn)行real-cugan有兩種壓制方法。本篇先講傳統(tǒng)的走pytorch的real-cugan的視頻壓制模式。后續(xù)有更效率更快速的配置模式。
系列一:
Vapoursynth視頻壓制教程 系列一:相關(guān)文件下載與環(huán)境配置 - 嗶哩嗶哩 (bilibili.com)
如果你啥環(huán)境都還沒裝,可以參考上面的鏈接搭建完Vapoursynth和python的運(yùn)行環(huán)境。如果你是按上述教程的搭建的環(huán)境的話可以直接參考下面給的教程完成本篇的內(nèi)容。如果不是的話可能需要自行對某些步驟進(jìn)行調(diào)整修改。
以下分為三部分
一,pytorch的安裝
二,real-cugan文件配置
三,測試
下面的文件安裝路徑,文件內(nèi)容等都是基于Vapoursynth視頻壓制教程 系列一:相關(guān)文件下載與環(huán)境配置 - 嗶哩嗶哩 (bilibili.com)這篇專欄內(nèi)容搭建的環(huán)境進(jìn)行測試。
一,pytorch的安裝
打開pytorch的官網(wǎng):https://pytorch.org/

選擇好對應(yīng)的安裝方式和Cuda版本,然后把藍(lán)色框里面的這串代碼復(fù)制下來,一會要用。
打開CMD輸入這串代碼

回車等待安裝完成

pytorch安裝好以后重啟一下VapourSynth Editor
二,real-cugan文件配置
來到:https://github.com/bilibili/ailab/tree/main/Real-CUGAN/VapourSynth
下載圖中紅色邊框標(biāo)注的文件upcunet_v20220227_vs.py

這個(gè)文件是適配Vapoursynth的real-cugan的推理代碼,但在這篇教程編寫的時(shí)候還不支持Pro版本的模型,可以到Pull requests里面查看相關(guān)方法修改這個(gè)文件來支持運(yùn)行Pro模型。
把這個(gè)文件放到python安裝目錄的Lib\site-packages

upcunet_v20220227_vs.py文件存放路徑
把real-cugan的一系列pth格式的模型存放到一個(gè)文件夾里,后續(xù)需要指定路徑調(diào)用。

三,測試
簡單寫一個(gè)測試腳本,這里假設(shè)插件和腳本環(huán)境已經(jīng)在系列一中按教程裝好。


在test.vpy的同目錄下打開命令行CMD,輸入簡單的ffmpeg管道壓制測試代碼:
vspipe?--y4m?test.vpy?-?|?ffmpeg?-i?pipe:?test.mp4


等待CMD跑完代碼就行,同目錄會有test.mp4的相關(guān)文件。
總結(jié):
上述測試均在本地電腦和云端電腦上成功運(yùn)行。
1080超分到4K
3060ti,不開半精度,超分辨率+壓制速度是一秒1.1幀。
3060ti,打開半精度,超分辨率+壓制速度是一秒2.1幀。
這里不考慮CPU壓制瓶頸。
本地測試電腦太渣,結(jié)果參差不齊就不展示結(jié)果了。
后續(xù)不會出視頻教程,請不要來私信問會不會出視頻教程,自己沒有設(shè)備對配音也不自信。配置過程中有什么問題可以在評論底下問,我能解決的話會盡量幫忙解決。三連和關(guān)注的同學(xué)提的問題優(yōu)先處理。純白嫖的同學(xué)提問題看情況隨緣解決。(滑稽)