Browse Source
Process generator input
Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
main
1 changed files with
2 additions and
1 deletions
-
isc.py
|
|
@ -69,10 +69,11 @@ class Isc(NNOperator): |
|
|
]) |
|
|
]) |
|
|
|
|
|
|
|
|
def __call__(self, data: Union[List[towhee._types.Image], towhee._types.Image]): |
|
|
def __call__(self, data: Union[List[towhee._types.Image], towhee._types.Image]): |
|
|
if not isinstance(data, list): |
|
|
|
|
|
|
|
|
if isinstance(data, towhee._types.Image): |
|
|
imgs = [data] |
|
|
imgs = [data] |
|
|
else: |
|
|
else: |
|
|
imgs = data |
|
|
imgs = data |
|
|
|
|
|
|
|
|
img_list = [] |
|
|
img_list = [] |
|
|
for img in imgs: |
|
|
for img in imgs: |
|
|
img = self.convert_img(img) |
|
|
img = self.convert_img(img) |
|
|
|
