Solved by ptrblck in post #2 
              
              
                Do you want to load all images from the volume at once and return them? 
I think the best approach would be to create a custom Dataset and implement the desired logic there. 
Basically you could pass the volume paths to __init__, and load all images corresponding to a volume in __getitem__. 
Have a …