From 74c7cc3175761d475b620a4c8f3b3e19a73421f6 Mon Sep 17 00:00:00 2001 From: macro chen Date: Thu, 5 Dec 2024 17:56:18 +0800 Subject: [PATCH] fix bug of ImageMaskScaleAs --- py/image_mask_scale_as.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/image_mask_scale_as.py b/py/image_mask_scale_as.py index 19948ee..0638266 100644 --- a/py/image_mask_scale_as.py +++ b/py/image_mask_scale_as.py @@ -1,6 +1,6 @@ import torch from PIL import Image -from .imagefunc import AnyType, log, tensor2pil, pil2tensor, fit_resize_image +from .imagefunc import AnyType, log, tensor2pil, pil2tensor, image2mask, fit_resize_image