logo
Browse Source

Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 3 years ago
parent
commit
b4eb0b91eb
  1. 8
      transform_image.py

8
transform_image.py

@ -54,9 +54,9 @@ class TransformImage(Operator):
(`torch.Tensor`)
The normalized image tensor.
"""
if isinstance(img, str):
img_tensor = Image.open(img)
if isinstance(img, Image.Image):
img_tensor = img.convert('RGB')
if isinstance(img_path, str):
img_tensor = Image.open(img_path)
if isinstance(img_path, Image.Image):
img_tensor = img_path.convert('RGB')
Outputs = NamedTuple('Outputs', [('img_transformed', torch.Tensor)])
return Outputs(self.tfms(img_tensor).unsqueeze(0))

Loading…
Cancel
Save