Refactor NiftiDataLoader to return transposed numpy array from NIfTI image

Signed-off-by: Simone Bendazzoli <[email protected]>

Author: SimoneBendazzoli93

monai/deploy/operators/nii_data_loader_operator.py

@@ -80,7 +80,8 @@ def convert_and_save(self, nii_path):
         image_reader = SimpleITK.ImageFileReader()
         image_reader.SetFileName(str(nii_path))
         image = image_reader.Execute()
-        return image
+        image_np = np.transpose(SimpleITK.GetArrayFromImage(image), [2, 1, 0])
+        return image_np
 
 
 def test():