From 3e634f76dfe1cd4e73572887527f71959dcea2d0 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sun, 5 Apr 2020 23:16:00 +0200
Subject: [PATCH] Add setter for projection_matrices
---
src/pyronn_torch/conebeam.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/src/pyronn_torch/conebeam.py b/src/pyronn_torch/conebeam.py
index aa0819e..816845d 100644
--- a/src/pyronn_torch/conebeam.py
+++ b/src/pyronn_torch/conebeam.py
@@ -170,3 +170,12 @@ class ConeBeamProjector:
self._inverse_matrices = torch.stack(tuple(map(torch.from_numpy, inv_matrices))).cuda().contiguous()
self._source_points = torch.stack(tuple(map(torch.from_numpy, source_points))).cuda().contiguous()
self._projection_multiplier = 1.
+
+ @property
+ def projection_matrices(self):
+ return self._projection_matrices_numpy
+
+ @projection_matrices.setter
+ def projection_matrices(self, numpy_matrices):
+ self._projection_matrices_numpy = numpy_matrices
+ self._calc_inverse_matrices()
--
GitLab