From 08a962cdfb0067896bfc3a4e295f426608fcb055 Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 27 Jun 2023 10:08:36 +0800 Subject: [PATCH] 1.2.1 --- lama_cleaner/server.py | 13 +++++++++++-- setup.py | 2 +- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py index c186ce2..755dc62 100644 --- a/lama_cleaner/server.py +++ b/lama_cleaner/server.py @@ -138,11 +138,20 @@ def save_image(): filename = request.form["filename"] origin_image_bytes = input["image"].read() # RGB ext = get_image_ext(origin_image_bytes) - image, _, exif_infos = load_img(origin_image_bytes, return_exif=True) + image, alpha_channel, exif_infos = load_img(origin_image_bytes, return_exif=True) save_path = os.path.join(output_dir, filename) + if alpha_channel is not None: + if alpha_channel.shape[:2] != image.shape[:2]: + alpha_channel = cv2.resize( + alpha_channel, dsize=(image.shape[1], image.shape[0]) + ) + image = np.concatenate((image, alpha_channel[:, :, np.newaxis]), axis=-1) + + pil_image = Image.fromarray(image) + img_bytes = pil_to_bytes( - Image.fromarray(image), + pil_image, ext, quality=image_quality, exif_infos=exif_infos, diff --git a/setup.py b/setup.py index 20fad90..5c9cd3b 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,7 @@ def load_requirements(): # https://setuptools.readthedocs.io/en/latest/setuptools.html#including-data-files setuptools.setup( name="lama-cleaner", - version="1.2.0", + version="1.2.1", author="PanicByte", author_email="cwq1913@gmail.com", description="Image inpainting tool powered by SOTA AI Model",