image-decode/image_decode.py

# encoding=utf-8

from towhee import register, ops
from towhee.operator import Operator


@register(output_schema='image')
class ImageDecode(Operator):

    def __init__(self, *args, **kwargs):
        self._op = ops.image_decode.cv2(*args, **kwargs)

    def __call__(self, image_path: str):
        return self._op(image_path)