diff options
| author | Edoardo Pasca <edo.paskino@gmail.com> | 2019-12-17 13:17:07 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-12-17 13:17:07 +0000 |
| commit | 76374867549ee85e89478ce894875ca27217f836 (patch) | |
| tree | 67695949a4d904b89b70c4257c30471622ea9590 | |
| parent | cdca2ba12a9dd3df67a00d0d3f40497aacadbe2b (diff) | |
| download | astra-wrapper-76374867549ee85e89478ce894875ca27217f836.tar.gz astra-wrapper-76374867549ee85e89478ce894875ca27217f836.tar.bz2 astra-wrapper-76374867549ee85e89478ce894875ca27217f836.tar.xz astra-wrapper-76374867549ee85e89478ce894875ca27217f836.zip | |
deallocates memory on GPU
closes #47
| -rw-r--r-- | Wrappers/Python/ccpi/astra/processors/FBP.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Wrappers/Python/ccpi/astra/processors/FBP.py b/Wrappers/Python/ccpi/astra/processors/FBP.py index adc4119..adf27eb 100644 --- a/Wrappers/Python/ccpi/astra/processors/FBP.py +++ b/Wrappers/Python/ccpi/astra/processors/FBP.py @@ -242,9 +242,10 @@ class FBP(DataProcessor): # since we run gpu we need to rescale and flip it IM.array[i] = np.flip(astra.data2d.get(rec_id) * self.volume_geometry.voxel_size_x,0) - astra.algorithm.delete(alg_id) + astra.algorithm.delete(alg_id) + astra.data2d.delete(sinogram_id) + astra.data2d.delete(rec_id) - astra.data2d.delete(sinogram_id) return IM |