GitHub |
Simple installation from pypi:
pip install tfkbnufft
This package is a verly early-stage and modest adaptation to TensorFlow of the torchkbnufft package written by Matthew Muckley for PyTorch. Please cite his work appropriately if you use this package.
The computation speeds are given in seconds, for a 256x256 image with a spokelength of 512 and 405 spokes. These numbers are not to be directly compared to those of torchkbnufft, since the computation is not the same. They are just to give a sense of the time required for computation.
Operation | CPU | GPU |
---|---|---|
Forward NUFFT | 0.1676 | 0.0626 |
Adjoint NUFFT | 0.7005 | 0.0635 |
To obtain these numbers for your machine, run the following commands, after installing this package:
pip install scikit-image Pillow
python profile_tfkbnufft.py
These numbers were obtained with a Quadro P5000.
-
Fessler, J. A., & Sutton, B. P. (2003). Nonuniform fast Fourier transforms using min-max interpolation. IEEE transactions on signal processing, 51(2), 560-574.
-
Beatty, P. J., Nishimura, D. G., & Pauly, J. M. (2005). Rapid gridding reconstruction with a minimal oversampling ratio. IEEE transactions on medical imaging, 24(6), 799-808.
-
Feichtinger, H. G., Gr, K., & Strohmer, T. (1995). Efficient numerical methods in non-uniform sampling theory. Numerische Mathematik, 69(4), 423-440.
If you want to cite the package, you can use any of the following:
@conference{muckley:20:tah,
author = {M. J. Muckley and R. Stern and T. Murrell and F. Knoll},
title = {{TorchKbNufft}: A High-Level, Hardware-Agnostic Non-Uniform Fast Fourier Transform},
booktitle = {ISMRM Workshop on Data Sampling \& Image Reconstruction},
year = 2020
}
@misc{Muckley2019,
author = {Muckley, M.J. et al.},
title = {Torch KB-NUFFT},
year = {2019},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/mmuckley/torchkbnufft}}
}