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