OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-29 16:09:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed upscaler test

Browse Source
This commit is contained in:
k4yt3x 2023-05-31 15:23:53 +00:00
parent 155a144a93
commit d66b7fa144
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_upscaler.py
View File

@ -29,16 +29,11 @@ def test_upscaling():
def test_upscale_image(): def test_upscale_image():
# initialize upscaler instance # initialize upscaler instance
processing_queue = multiprocessing.Queue(maxsize=30) upscaler = Upscaler()
processed_frames = multiprocessing.Manager().list([None])
pause = multiprocessing.Value(ctypes.c_bool, False)
upscaler = Upscaler(processing_queue, processed_frames, pause)
image = Image.open("data/test_image.png") image = Image.open("data/test_image.png")
upscaled_image = upscaler.upscale_image(image, 1680, 960, "waifu2x", 3) upscaled_image = upscaler.upscale_image(image, 1680, 960, "waifu2x", 3)
reference_image = Image.open("data/test_image_ref.png") reference_image = Image.open("data/test_image_ref.png")
assert utils.get_image_diff(upscaled_image, reference_image) < 0.5 assert utils.get_image_diff(upscaled_image, reference_image) < 0.5