From b4eb0b91eb7d1678b056419073a429723a0486e2 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Thu, 16 Dec 2021 17:16:17 +0800 Subject: [PATCH] Update Signed-off-by: shiyu22 --- transform_image.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/transform_image.py b/transform_image.py index f2fd873..5d5ed8e 100644 --- a/transform_image.py +++ b/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))