Quick toy example for segmentation

inference.py

import torchio as tio

# Loading the data and the expected labels
subject = tio.Subject(
    mri=tio.ScalarImage("./T2w_hires_right_hippocampus.nii.gz"),
    label=tio.LabelMap("./T2w_hires_right_hippocampus_seg_crop.nii.gz"))

# Quick preprocessing to get the data in the right format for inference
preprocess = tio.Compose([
    tio.ZNormalization(),
    tio.EnsureShapeMultiple(8),
])
subject = preprocess(subject)

# Plot if needed
subject.plot()

model = dummy_model()
segmentation = model(subject.mri.data)