Is the code able to handle 3D data?

[txusser]

Hi,

I have been trying to do a slight modification to your code to work with Nifti images. I have been able to implement the image load with nibabel.

My data are paired 3D DatSPECT images of 111134102. Nevertheless, it seems that the code is interpreting my data as 2D data with 111 channels, which seems incorrect. The same issue appears when I store the data as .npy

Is the code capable of handling this type of 3D medical images?