Have you solved this problem?
Since each 3D image is very large , I also need to choose some 2D slices from 3D image.
I think we can write the path of 2D slice and read them in __getitem() function…
Do you have another better solution??