Spaces:
Sleeping
Sleeping
| import nibabel as nib | |
| import numpy as np | |
| def load_ct_to_numpy(data_path): | |
| if not isinstance(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 < -1024] = 1024 | |
| data[data > 1024] = 1024 | |
| 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 not isinstance(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])] | |