`color_mode='gray'` should return `(height, width, 1)` instead of `(height, width)`

mjkvaak / ImageDataAugmentor

Custom image data generator for TF Keras that supports the modern augmentation module albumentations

Other

86 stars 27 forks source link

`color_mode='gray'` should return `(height, width, 1)` instead of `(height, width)` #4

Closed mjkvaak closed 4 years ago

mjkvaak commented 4 years ago

I suppose the simple solution is just to add a line img = img[..., np.newaxis] somewhere before the batches are being output in casecolor_mode='gray'.

mjkvaak commented 4 years ago

Fixed