logo
Browse Source

rm _types

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
junjie.jiang 1 year ago
parent
commit
552b05d807
  1. 2
      swag.py

2
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)

Loading…
Cancel
Save