Question about MRI2DSegmentationDataset Coronal and Sagittal view

[MohitTare]

While exploring the code i saw that when we are trying to create a dataloader for Coronal and Sagittal view of the NIFTI image we get same dataset length as the axial view (slice_axis=2).

If we wanted to plot a sagittal view directly from NIFTI image we would usually do the following in base numpy

sg = np.transpose(img_data, [1, 2,0])
sg = np.rot90(sg, 1)

The plot the particular sagittal slice with nth slice number

plt.imshow(sg[:,:,n], cmap='gray',aspect = 5)

But i am not sure how we can have this with current implementation of dataset. Wont the dataset have a different length (x dimension in this case). Please let me know if my understanding is correct on this or in case i am missing something out