import nibabel as nib
import numpy as np
path='../data/'
fileName='wr_Lac_78.nii'
data = nib.load(path+fileName).get_fdata()
data = data.transpose(2, 0, 1)
data = np.nan_to_num(data)
print(np.shape(data))
import nibabel as nib
import numpy as np
path='../data/'
fileName='wr_Lac_78.nii'
data = nib.load(path+fileName).get_fdata()
data = data.transpose(2, 0, 1)
data = np.nan_to_num(data)
print(np.shape(data))