summaryrefslogtreecommitdiffstats
path: root/src/Python/ccpi
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2017-10-27 12:40:53 +0100
committerEdoardo Pasca <edo.paskino@gmail.com>2017-10-27 12:40:53 +0100
commita9ee40156a7ad6ae4fcb9e07aea99914f6c7f7ff (patch)
treef56f689bc15de7e61bb080795453a850d194c711 /src/Python/ccpi
parent45ce9d604f2fe6f58318c8103f3f6125af9cb9bc (diff)
downloadregularization-a9ee40156a7ad6ae4fcb9e07aea99914f6c7f7ff.tar.gz
regularization-a9ee40156a7ad6ae4fcb9e07aea99914f6c7f7ff.tar.bz2
regularization-a9ee40156a7ad6ae4fcb9e07aea99914f6c7f7ff.tar.xz
regularization-a9ee40156a7ad6ae4fcb9e07aea99914f6c7f7ff.zip
handles correctly the slicing
Diffstat (limited to 'src/Python/ccpi')
-rw-r--r--src/Python/ccpi/imaging/Regularizer.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/Python/ccpi/imaging/Regularizer.py b/src/Python/ccpi/imaging/Regularizer.py
index fb9ae08..8ab6c6a 100644
--- a/src/Python/ccpi/imaging/Regularizer.py
+++ b/src/Python/ccpi/imaging/Regularizer.py
@@ -216,13 +216,13 @@ class Regularizer():
#assuming it's 3D
# run independent calls on each slice
out3d = input.copy()
- for i in range(np.shape(input)[2]):
- out = self.algorithm(input, regularization_parameter,
+ for i in range(np.shape(input)[0]):
+ out = self.algorithm(input[i], regularization_parameter,
self.pars['first_order_term'] ,
self.pars['second_order_term'] ,
self.pars['number_of_iterations'])
# copy the result in the 3D image
- out3d.T[i] = out[0].copy()
+ out3d[i] = out[0].copy()
# append the rest of the info that the algorithm returns
output = [out3d]
for i in range(1,len(out)):