From 30b29db6683fd98b342f7a2e970e324eea701a82 Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 30 Apr 2024 13:13:13 +0800 Subject: [PATCH] fix paint_by_example test --- iopaint/tests/test_paint_by_example.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/iopaint/tests/test_paint_by_example.py b/iopaint/tests/test_paint_by_example.py index 27b8a77..9447138 100644 --- a/iopaint/tests/test_paint_by_example.py +++ b/iopaint/tests/test_paint_by_example.py @@ -1,6 +1,7 @@ import cv2 import pytest from PIL import Image +from iopaint.helper import encode_pil_to_base64 from iopaint.model_manager import ModelManager from iopaint.schema import HDStrategy @@ -34,7 +35,9 @@ def assert_equal( ) print(f"Input image shape: {img.shape}, example_image: {example_image.shape}") - config.paint_by_example_example_image = Image.fromarray(example_image) + config.paint_by_example_example_image = encode_pil_to_base64( + Image.fromarray(example_image), 100, {} + ).decode("utf-8") res = model(img, mask, config) cv2.imwrite(str(save_dir / save_name), res)