From e4374dccef84235af32124a7860b81d53fe76314 Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Thu, 4 Aug 2022 18:15:15 +0800 Subject: [PATCH] add Signed-off-by: junjie.jiang --- __init__.py | 20 ++++++++++++++++++++ image_decoder.py | 5 ----- 2 files changed, 20 insertions(+), 5 deletions(-) create mode 100644 __init__.py diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..1bf901e --- /dev/null +++ b/__init__.py @@ -0,0 +1,20 @@ +# Copyright 2021 Zilliz. All rights reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from .image_decoder import ImageDecoder + + +# The factory method +def nvjpeg(): + return ImageDecoder() diff --git a/image_decoder.py b/image_decoder.py index 57c14fa..1f48070 100644 --- a/image_decoder.py +++ b/image_decoder.py @@ -89,8 +89,3 @@ class ImageDecoder(PyOperator): @property def shared_type(self): return SharedType.NotShareable - -if __name__ == '__main__': - d = ImageDecoder() - print(d('/home/junjie.jiangjjj/images/1.png')) - print(d('/home/junjie.jiangjjj/images/towhee.jpeg'))