Skip to content

Repository files navigation

sr-vulkan

Install

pip install sr-vulkan -v

Install Model

pip install sr-vulkan-model-waifu2x
pip install sr-vulkan-model-realcugan
pip install sr-vulkan-model-realesrgan
pip install sr-vulkan-model-realsr

Use

from sr_vulkan import sr_vulkan as sr
# init
sr.setDebug(True)
sts = sr.init()
if sts < 0:
# cpu model
isCpuModel = True
gpuList = sr.getGpuInfo()
print(gpuList)
sts = sr.initSet(gpuId=0, threadNum=2)
assert sts==0
# add picture ...# sr.add(...)# load picture...# newData, status, backId, tick = sr.load(0)

Example

  • Please see test Example

Build

pip install wheel
python setup.py bdist_wheel

About

waifu2x-ncnn-vulkan-python, use nihui/waifu2x-ncnn-vulkan

Topics

Resources

Stars

57 stars

Watchers

1 watching

Forks

Releases

Used by

Contributors

Languages