Browse Source
rm _types
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
swag.py
|
|
@ -59,7 +59,7 @@ class Swag(NNOperator): |
|
|
|
self.model.eval() |
|
|
|
|
|
|
|
@arg(1, to_image_color('RGB')) |
|
|
|
def __call__(self, img: towhee._types.Image) -> numpy.ndarray: |
|
|
|
def __call__(self, img: 'towhee.types.Image') -> numpy.ndarray: |
|
|
|
img = PILImage.fromarray(img.astype('uint8'), 'RGB') |
|
|
|
if not self.skip_tfms: |
|
|
|
img = self.tfms(img).unsqueeze(0) |
|
|
|