diff options
| author | dkazanc <dkazanc@hotmail.com> | 2019-03-06 17:37:14 +0000 |
|---|---|---|
| committer | dkazanc <dkazanc@hotmail.com> | 2019-03-06 17:37:14 +0000 |
| commit | 4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92 (patch) | |
| tree | 356cae729e8d783adc8151174bef57b6feabd4e0 /src/Python | |
| parent | 39baef90c4b209090f006e5308653cb0a3348c4e (diff) | |
| download | regularization-4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92.tar.gz regularization-4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92.tar.bz2 regularization-4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92.tar.xz regularization-4b29a6adc924bf8a4b3e4f9835ded93a3a2f7b92.zip | |
fixes 2n vector problem
Diffstat (limited to 'src/Python')
| -rw-r--r-- | src/Python/src/cpu_regularisers.pyx | 6 |
1 files changed, 3 insertions, 3 deletions
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, |