ZFTurbo/volumentations

resize error

JianK1216 opened this issue · 1 comments

JianK1216 commented

Hi ZFTurbo,
Thanks very much for your great work！ but I encountered an error when resize the image. Many thanks for your help in advance.

the error is as below:
111

JianK1216 commented

I have solved this problem. This error is caused by the 1 channel input(16,32,32). The default input expects a 3D image in color mode(None, None,None,3). So we need to change the code on ../volumentations/augmentations.functionals.py in line 138 as below:
image

type =1 for color(3channel) mode input shape: (None, None, None, 3) ,
type=0 for grayscale(1 channel) mode input shape: (None, None, None)