Skip to content

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

License

tonquer/sr-vulkan

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

License

Stars

Watchers

Forks

Contributors 4

  •  
  •  
  •  
  •