Skip to content

Latest commit

 

History

History
49 lines (41 loc) · 1.14 KB

File metadata and controls

49 lines (41 loc) · 1.14 KB

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