resize error
JianK1216 opened this issue · comments
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:
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:
type =1 for color(3channel) mode input shape: (None, None, None, 3) ,
type=0 for grayscale(1 channel) mode input shape: (None, None, None)