From 042d4a4948372bd1c58e7afd1c08ff8b9f51a4e3 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Wed, 25 Nov 2020 17:20:48 +0800
Subject: [PATCH] Update get_dr_txt.py

---
 get_dr_txt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/get_dr_txt.py b/get_dr_txt.py
index 9c0547b..f92bcd0 100644
--- a/get_dr_txt.py
+++ b/get_dr_txt.py
@@ -28,7 +28,7 @@ class mAP_Yolo(YOLO):
         f = open("./input/detection-results/"+image_id+".txt","w") 
         image_shape = np.array(np.shape(image)[0:2])
 
-        crop_img = np.array(letterbox_image(image, (self.model_image_size[0],self.model_image_size[1])))
+        crop_img = np.array(letterbox_image(image, (self.model_image_size[1],self.model_image_size[0])))
         photo = np.array(crop_img,dtype = np.float32)
         photo /= 255.0
         photo = np.transpose(photo, (2, 0, 1))
-- 
GitLab