Loading large sized data

I am working on 114 3D images each having 3D segmentation labels. For each image, I can generate 96 or 192 3D patches. I want to know how to load the images in Dataloader since the patch from all the images can’t be loaded together.

  1. custom subclass of Dataset
  2. save file path in __init__
  3. load each image and generate its patches in __getitem__
1 Like