From 4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92 Mon Sep 17 00:00:00 2001 From: dkazanc Date: Wed, 6 Mar 2019 17:37:14 +0000 Subject: fixes 2n vector problem --- src/Python/src/cpu_regularisers.pyx | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/Python') diff --git a/src/Python/src/cpu_regularisers.pyx b/src/Python/src/cpu_regularisers.pyx index b7d029d..49cdf94 100644 --- a/src/Python/src/cpu_regularisers.pyx +++ b/src/Python/src/cpu_regularisers.pyx @@ -100,11 +100,11 @@ def TV_FGP_2D(np.ndarray[np.float32_t, ndim=2, mode="c"] inputData, cdef np.ndarray[np.float32_t, ndim=2, mode="c"] outputData = \ np.zeros([dims[0],dims[1]], dtype='float32') - cdef np.ndarray[np.float32_t, ndim=2, mode="c"] infovec = \ - np.ones([dims[0],dims[1]], dtype='float32') + cdef np.ndarray[np.float32_t, ndim=1, mode="c"] infovec = \ + np.ones([2], dtype='float32') #/* Run FGP-TV iterations for 2D data */ - TV_FGP_CPU_main(&inputData[0,0], &outputData[0,0], &infovec[0,0], regularisation_parameter, + TV_FGP_CPU_main(&inputData[0,0], &outputData[0,0], &infovec[0], regularisation_parameter, iterationsNumb, tolerance_param, methodTV, -- cgit v1.2.3