From 977674e72ef58574bfbe85aa92c99941812b2a43 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Tue, 16 Apr 2024 14:56:39 +0000
Subject: [PATCH] Correctly assess image size
---
atr_data_generator/extract/base.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/atr_data_generator/extract/base.py b/atr_data_generator/extract/base.py
index 7295e13..c899df1 100644
--- a/atr_data_generator/extract/base.py
+++ b/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),
--
GitLab