search

ZFTurbo / volumentations

Library for 3D augmentations
MIT License
220 stars 35 forks source link

resize error #4

Closed JianK1216 closed 2 years ago

JianK1216 commented 2 years ago

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 2 years ago

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)