segmentation_models_3D icon indicating copy to clipboard operation
segmentation_models_3D copied to clipboard

Unequal Dataset Dimension

Open mudabir94 opened this issue 4 years ago • 1 comments

I have a spectral cube dataset of shape (100,256,256,50). There are 100 images of 256 by 256 of 50 dimensions. I want to use such kind of data on the 3D network but according to my understanding the size of height, width and dimensions should be equal otherwise the model won't compile. Please guide what to do in this scenario. Thank you.

mudabir94 avatar Apr 12 '22 20:04 mudabir94

First 3 dimensions must be divisible on 32. So something like (96,256,256,50) will be ok. Also have in mind that it will require a lot of GPU memory. If you have 48 GB then it probably fit, otherwise I'm not sure.

ZFTurbo avatar Apr 12 '22 20:04 ZFTurbo