Hi all!
I would like to use a 3D U-Net model for segmentation but I am not sure how to create an appropriate 3D dataloader for the dataset. The full dataset is 240x240x155 and I would like to create Bx1x64x64x64  for example. I currently have a dataloader that can output the whole volume chunked up into 64x64x64 voxels but I am having trouble in randomizing the voxel volumes.
Does anyone have any suggestions? Is there a recommended pytorch approach to use?