File size: 842 Bytes
4e18454 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import nibabel as nib
import numpy as np
def load_ct_to_numpy(data_path):
if type(data_path) != str:
data_path = data_path.name
image = nib.load(data_path)
data = image.get_fdata()
data = np.rot90(data, k=1, axes=(0, 1))
data[data < -150] = -150
data[data > 250] = 250
data = data - np.amin(data)
data = data / np.amax(data) * 255
data = data.astype("uint8")
print(data.shape)
return [data[..., i] for i in range(data.shape[-1])]
def load_pred_volume_to_numpy(data_path):
if type(data_path) != str:
data_path = data_path.name
image = nib.load(data_path)
data = image.get_fdata()
data = np.rot90(data, k=1, axes=(0, 1))
data[data > 0] = 1
data = data.astype("uint8")
print(data.shape)
return [data[..., i] for i in range(data.shape[-1])]
|