search

mjkvaak / ImageDataAugmentor

Custom image data generator for TF Keras that supports the modern augmentation module albumentations
Other
86 stars 27 forks source link

`data_format` parameter is irresponsive #3

Closed mjkvaak closed 4 years ago

mjkvaak commented 4 years ago

The code uses cv2 which to my understanding processes data in (height, width, channels) format. In before returning the batches, the generator should swap the axes order to (channels, height, width) if data_format = 'channels_first'.

mjkvaak commented 4 years ago

Fixed