diff --git a/get_dr_txt.py b/get_dr_txt.py
index 9c0547bf1685ef635f4cac63a2109915e8ce081e..f92bcd0faea88b565f0d76f8cec54ee5a8b02e44 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))