Neuroimaging-Example/src/process_data.py

from .functions import *
from .const_process_data import *

t1, t2, flair, t1ce, seg = modalities_path(base_path)

assert len(t1) == len(t2) == len(flair) == len(t1ce) == len(seg), 'Missing modalities in some patients'
assert len(t1) == len(t2) == len(flair) == len(t1ce) == len(seg) > 0, 'The lists of paths are empty'

num_samples: int = len(t1)

data = load_raw_data(t1, t2, flair, t1ce, num_samples, input_shape)
save_data(t1, t2, flair, t1ce, data)

labels = load_raw_labels(seg, num_samples, input_shape, output_channels)
save_label(labels, seg, file_type)