Correctly assess image size

atr_data_generator/extract/base.py

@@ -123,7 +123,8 @@ class DataGenerator:
         )
 
         # Resize to required height if needed
-        if self.image.fixed_height and bbox.height != self.image.fixed_height:
+        # image.shape = (height, width)
+        if self.image.fixed_height and image.shape[0] != self.image.fixed_height:
             image = np.asarray(
                 resize_image_height(
                     img=Image.fromarray(image),