From 82254f587e14cdf911755d9f4110c175988aa03f Mon Sep 17 00:00:00 2001 From: chflame163 Date: Tue, 3 Sep 2024 11:46:30 +0800 Subject: [PATCH] fix SAM2Ultra bug --- py/sam_2_ultrl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/sam_2_ultrl.py b/py/sam_2_ultrl.py index 22875e3..35f957d 100644 --- a/py/sam_2_ultrl.py +++ b/py/sam_2_ultrl.py @@ -392,8 +392,8 @@ class LS_SAM2_ULTRA: max_megapixels=max_megapixels) _mask = tensor2pil(histogram_remap(pil2tensor(_mask), black_point, white_point)) else: - _mask = mask2image(_mask) - + _mask = tensor2pil(_mask) + print(f"orig_image.size={orig_image.size},_mask.size={_mask.size}") ret_image = RGB2RGBA(orig_image, _mask.convert('L')) ret_images.append(pil2tensor(ret_image)) ret_masks.append(image2mask(_mask))