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.
- Preprocessing routines for NSI and Zeiss scanners.
- Utilities for metal artifact reduction and stripe removal.
- 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 mbirtorchReconstruct in one line:
import mbirtorch
recon, recon_dict = mbirtorch.recon_simple_parallel(sinogram, angles)Full documentation at https://mbirtorch.readthedocs.io/