You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
307 B
10 lines
307 B
# -*- 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
|