Skip to content
Snippets Groups Projects
Commit 977674e7 authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Correctly assess image size

parent ff970bf9
No related branches found
No related tags found
1 merge request!94Correctly assess image size
Pipeline #175281 passed
......@@ -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),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment