From 4cc486b8070a9bb2a057205d7b4e1519e3b3daff Mon Sep 17 00:00:00 2001 From: "junjie.jiangjjj" Date: Tue, 28 Jun 2022 11:30:13 +0800 Subject: [PATCH] update Signed-off-by: junjie.jiangjjj --- __init__.py | 4 ++-- image_decode_cv2.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/__init__.py b/__init__.py index 3254229..eec8e9c 100644 --- a/__init__.py +++ b/__init__.py @@ -12,9 +12,9 @@ # See the License for the specific language governing permissions and # limitations under the License. -from .image_decode_cv2 import ImageDecodeCV2 +from .image_decode_cv2 import ImageDecoder # The factory method def test_cv2(): - return ImageDecodeCV2() + return ImageDecoder() diff --git a/image_decode_cv2.py b/image_decode_cv2.py index b6a6012..70bcc37 100644 --- a/image_decode_cv2.py +++ b/image_decode_cv2.py @@ -6,7 +6,7 @@ import cv2 import requests import numpy as np -from towhee.types import Image +from towhee._types import Image from towhee.operator import PyOperator, SharedType from towhee import register from towhee.utils import ndarray_utils