diff --git a/preprocessing.py b/preprocessing.py index a66453102..50db682bf 100644 --- a/preprocessing.py +++ b/preprocessing.py @@ -278,7 +278,7 @@ def aug_image(self, train_instance, jitter): image = self.aug_pipe.augment_image(image) # resize the image to standard size - image = cv2.resize(image, (self.config['IMAGE_H'], self.config['IMAGE_W'])) + image = cv2.resize(image, (self.config['IMAGE_W'], self.config['IMAGE_H'])) image = image[:,:,::-1] # fix object's position and size