From 856c4323b7fdc01f078fc2aad43879a1219afb97 Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Mon, 25 Jul 2022 19:51:45 +0800 Subject: [PATCH] [to #43259593] fix typo --- modelscope/pipelines/cv/image_colorization_pipeline.py | 2 +- modelscope/pipelines/cv/image_matting_pipeline.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/pipelines/cv/image_colorization_pipeline.py b/modelscope/pipelines/cv/image_colorization_pipeline.py index 8992ba8e..cac6a344 100644 --- a/modelscope/pipelines/cv/image_colorization_pipeline.py +++ b/modelscope/pipelines/cv/image_colorization_pipeline.py @@ -88,7 +88,7 @@ class ImageColorizationPipeline(Pipeline): img = input.convert('LA').convert('RGB') elif isinstance(input, np.ndarray): if len(input.shape) == 2: - img = cv2.cvtColor(input, cv2.COLOR_GRAY2BGR) + input = cv2.cvtColor(input, cv2.COLOR_GRAY2BGR) img = input[:, :, ::-1] # in rgb order img = PIL.Image.fromarray(img).convert('LA').convert('RGB') else: diff --git a/modelscope/pipelines/cv/image_matting_pipeline.py b/modelscope/pipelines/cv/image_matting_pipeline.py index 691b1d94..3166c9a8 100644 --- a/modelscope/pipelines/cv/image_matting_pipeline.py +++ b/modelscope/pipelines/cv/image_matting_pipeline.py @@ -46,7 +46,7 @@ class ImageMattingPipeline(Pipeline): logger.info('load model done') def preprocess(self, input: Input) -> Dict[str, Any]: - img = LoadImage.convert_to_img(input) + img = LoadImage.convert_to_ndarray(input) img = img.astype(np.float) result = {'img': img} return result