LLNL / LEAP

comprehensive library of 3D transmission Computed Tomography (CT) algorithms with Python API and fully integrated with PyTorch
https://leapct.readthedocs.io
MIT License
74 stars 9 forks source link

Total Variation functions do not work for "large" arrays #5

Closed kylechampley closed 6 months ago

kylechampley commented 7 months ago

The Total Variation (TV) functions (cost, gradient quadratic form) do not work for volumes larger than about 1K^3. We are currently working on implementing these algorithms so that they split up the volume into smaller chunks that can be processed on a GPU. We will also make these function compute using multiple GPUs while we are at it.