Resample only one axis

In this example, we create a custom preprocessing transfom that changes the image spacing across one axis only.

Inspired by this discussion.

Downloading https://github.com/fepegar/torchio-data/raw/main/data/fernando/t1.nii.gz to /home/docs/.cache/torchio/fpg/t1.nii.gz

0it [00:00, ?it/s]
  0%|          | 0/10860389 [00:00<?, ?it/s]
10862592it [00:00, 15439857.76it/s]
Downloading https://github.com/fepegar/torchio-data/raw/main/data/fernando/t1_seg_gif.nii.gz to /home/docs/.cache/torchio/fpg/t1_seg_gif.nii.gz

0it [00:00, ?it/s]
  0%|          | 0/544126 [00:00<?, ?it/s]
548864it [00:00, 1251181.41it/s]
Downloading https://github.com/fepegar/torchio-data/raw/main/data/fernando/t1_to_mni.tfm to /home/docs/.cache/torchio/fpg/t1_to_mni.tfm

0it [00:00, ?it/s]
  0%|          | 0/329 [00:00<?, ?it/s]
8192it [00:00, 23236.20it/s]
Downloading https://github.com/fepegar/torchio-data/raw/main/data/fernando/t1_to_mni_affine.h5 to /home/docs/.cache/torchio/fpg/t1_to_mni_affine.h5

0it [00:00, ?it/s]
  0%|          | 0/8392 [00:00<?, ?it/s]
16384it [00:00, 39177.27it/s]

import torch
import torchio as tio


class ResampleZ:
    def __init__(self, spacing_z):
        self.spacing_z = spacing_z

    def __call__(self, subject):
        # We'll assume all images in the subject have the same spacing
        sx, sy, _ = subject.spacing
        resample = tio.Resample((sx, sy, self.spacing_z))
        resampled = resample(subject)
        return resampled


torch.manual_seed(42)
image = tio.datasets.FPG().t1
transforms = tio.ToCanonical(), ResampleZ(spacing_z=7)
transform = tio.Compose(transforms)
transformed = transform(image)
subject = tio.Subject(original=image, transformed=transformed)
subject.plot()