diff --git a/inference_single_image.py b/inference_single_image.py
index 4c47275..a7a227e 100644
--- 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