Skip to content

Image Analysis Primitives#

These packages for image analysis help you to explore your new ideas without reinventing the wheel. Packages are implemented in PyTorch, this means they can run on GPUs and make use of autodiff for parameter optimisation.

Fourier Space#

  • torch-fourier-slice | extracting/inserting central slices of Fourier transforms |
  • torch-fourier-rescale | rescale by padding/cropping Fourier transforms |
  • torch-fourier-shift | subpixel shift by phase shifting Fourier transforms |
  • torch-fourier-filter | Fourier space filters (including the CTF) |
  • torch-fourier-shell-correlation | correlation as a function of spatial frequency |

Real Space#

  • torch-image-interpolation | sample values from or insert values into images |
  • torch-transform-image | affine transforms of images |
  • torch-cubic-spline-grids | continuous parametrisations of 1-4D spaces |
  • torch-subpixel-crop | crop from images with subpixel precision |
  • torch-find-peaks | find and refine peaks in images |

Utilities#

  • torch-grid-utils | coordinate grids, frequency grids and shape generation |
  • torch-so3 | 3D rotation operations and utilities |
  • torch-affine-utils | affine matrix generation for 2D/3D coordinates |