Scaling 3D images in pytorch

abweiss · May 23, 2017, 6:59pm

As a preprocessing step, I need to scale 3D images. Does pytorch have a 3D bilinear interpolation tool or any other useful upsample/downsample tools for this purpose?

alexis-jacq · May 24, 2017, 8:19pm

Torchvision.transforms.Scale is based on Pil’s tools for interpolation (Pil.image.resize) that works in 2D only. To your purpose, you should rather have a look to scipy ndimages : https://docs.scipy.org/doc/scipy/reference/ndimage.html

ajbrock · May 25, 2017, 11:19am

I wrote a pytorch GPU-accelerated 3d resampler a while back that uses catmull-rom spline interpolation

gist.github.com

https://gist.github.com/ajbrock/86fd41cf468839fe804ce28df145d170

resample3d.py

# Resample.py
# Andrew Brock, 2017
# This code resamples a 3d grid using catmull-rom spline interpolation, and is GPU accelerated.

# Resample along the trailing dimension
# Assumes a more-than-1D array? Or just directly assumes a 3D array? we'll find out
# 

# TODO: Some things could be shared (such as the mgrid call, which can presumably be done once? hmm)
#       between resample1d calls.

This file has been truncated. show original