fixed

svip-lab · Nov 7, 2022 · efc8cdc · efc8cdc
1 parent 0d682b7
commit efc8cdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference_single_image.py b/inference_single_image.py
@@ -77,7 +77,7 @@ def inference(args):
         input_image = pil.open(image_path).convert('RGB')
         extension = image_path.split('.')[-1]
         original_width, original_height = input_image.size
-        input_image = input_image.resize((thisH, thisW), pil.LANCZOS)
+        input_image = input_image.resize((thisW, thisH), pil.LANCZOS)
         input_image = transforms.ToTensor()(input_image).unsqueeze(0)
 
         # PREDICTION