.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/plot_custom_z_spacing.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_plot_custom_z_spacing.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_plot_custom_z_spacing.py:


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 <https://github.com/TorchIO-project/torchio/discussions/885>`_.

.. GENERATED FROM PYTHON SOURCE LINES 11-36



.. image-sg:: /auto_examples/images/sphx_glr_plot_custom_z_spacing_001.png
   :alt: original (sagittal), original (coronal), original (axial), transformed (sagittal), transformed (coronal), transformed (axial)
   :srcset: /auto_examples/images/sphx_glr_plot_custom_z_spacing_001.png
   :class: sphx-glr-single-img


.. rst-class:: sphx-glr-script-out

 .. code-block:: none

    Downloading https://github.com/TorchIO-project/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, 33508481.43it/s]         
    Downloading https://github.com/TorchIO-project/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, 2085778.08it/s]          
    Downloading https://github.com/TorchIO-project/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, 42360.18it/s]           
    Downloading https://github.com/TorchIO-project/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, 58597.07it/s]           






|

.. code-block:: Python


    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()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 2.426 seconds)


.. _sphx_glr_download_auto_examples_plot_custom_z_spacing.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: plot_custom_z_spacing.ipynb <plot_custom_z_spacing.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: plot_custom_z_spacing.py <plot_custom_z_spacing.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: plot_custom_z_spacing.zip <plot_custom_z_spacing.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_