vs-mlrt - VapourSynth ML filter runtimes.

Selur

Hi,

dachte eventuell ist es für jemanden interessant: https://github.com/AmusementClub/vs-mlrt

Aktuell erlaubt es:

über die CPU zu ohne CUDA zu nutzen.

Beispiel Code für die Nutzung:

Code

# adjusting color space from YUV420P8 to RGBS for MLRT
clip = core.resize.Bicubic(clip=clip, format=vs.RGBS, matrix_in_s="470bg", range_s="limited")
sigma = 2.0
clipSigma = core.std.BlankClip(clip, format=vs.GRAYS, color=sigma/255.0)
clip = core.ov.Model(clips=[clip, clipSigma], network_path="I:/Hybrid/64bit/vsfilters/MiscFilter/vsMLRT/models/dpir/drunet_color.onnx")

Cu Selzr

Selur

... gibt jetzt auch CUDA support.

Selur

Dachte ich schick mal ein kleines Update.

Mittlerweile ist das Ganze recht einfach aufzusetzen.

Einfach vsmlrt-windows-x64-cuda.v9.7z herunterladen, in einen Ordner entpacken ( bei mir 'i:/Hybrid/64bit/vsfilters/vs-mlrt')

und dann kann man im Skript: Waifu2x, DPIR, RealESRGANv2 und CUGAN, verwenden:

Python

# Bibliothek lade
core.std.LoadPlugin(path="i:/Hybrid/64bit/vsfilters/vs-mlrt/vsort.dll")
# Wrapper laden
import imp
vsmlrt = imp.load_source('vsmlrt', 'i:/Hybrid/64bit/vsfilters/vs-mlrt/vsmlrt.py')
from vsmlrt import Backend


## Waifu2x:
#from vsmlrt import Waifu2x, Waifu2xModel
#clip = Waifu2x(clip,noise=-1, scale=2, model=Waifu2xModel.upconv_7_anime_style_art_rgb, backend=Backend.ORT_CUDA)


## DPIR
#from vsmlrt import DPIR, DPIRModel
#clip = DPIR(clip, strength=5, model=DPIRModel.drunet_color,backend=Backend.ORT_CUDA)


## RealESRGAN
#from vsmlrt import RealESRGANv2, RealESRGANv2Model
#clip = RealESRGANv2(clip, model=RealESRGANv2Model.animevideo_xsx2, backend=Backend.ORT_CUDA)


## CUGAN
#from vsmlrt import CUGAN
#clip = CUGAN(clip, noise=-1, scale=2, backend=Backend.ORT_CUDA)

Alles anzeigen

Cu Selur

Jetzt mitmachen!