3d dataloader for segmentation - vision - PyTorch Forums
discuss.pytorch.org › t › 3d-dataloader-forJun 20, 2019 · 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 ...