mbirtorch

MBIRTorch: Model-Based Iterative Reconstruction (MBIR) for tomographic reconstruction using PyTorch.

Features include:

• Multiple geometries: parallel beam, cone beam (including curved detector and helical), translation mode, and multi-axis parallel.
• 4D reconstruction: a time sequence of volumes from a single continuous scan, using multi-agent consensus equilibrium (MACE).
• Preprocessing routines for NSI and Zeiss scanners, plus geometry calibration.
• Utilities for metal artifact reduction and stripe removal.
• Interactive slice and geometry viewers.
• Informative demos and extensive documentation.
• Seamless operation on 1 or more GPUs, Mac MPS, or CPU.
• Compiled torch and Triton kernels for efficiency.

Available on PyPI via

pip install mbirtorch

Reconstruct in one line:

import mbirtorch
recon, recon_dict = mbirtorch.recon_simple_parallel(sinogram, angles)

Full documentation at https://mbirtorch.readthedocs.io/