Browse Source

add super resolution

pull/1/head
Alex Shonenkov 5 years ago
parent
commit
e3ad330f27
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      tests/test_super_resolution.py

+ 10
- 0
tests/test_super_resolution.py View File

@ -0,0 +1,10 @@
# -*- coding: utf-8 -*-
from rudalle.pipelines import super_resolution
def test_super_resolution(sample_image, realesrgan):
img = sample_image.copy()
img = img.resize((32, 32))
sr_img = super_resolution([img], realesrgan)[0]
assert sr_img.size[0] == 32*2
assert sr_img.size[1] == 32*2

Loading…
Cancel
Save