From c4f50db4f5b318aad785ae577908d37fe05f53d2 Mon Sep 17 00:00:00 2001 From: algol Date: Tue, 1 May 2018 15:30:28 +0100 Subject: some updates in demo --- Wrappers/Python/demos/demo_cpu_inpainters.py | 2 +- Wrappers/Python/src/cpu_regularisers.pyx | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'Wrappers') diff --git a/Wrappers/Python/demos/demo_cpu_inpainters.py b/Wrappers/Python/demos/demo_cpu_inpainters.py index ab7ed2f..9a677c4 100644 --- a/Wrappers/Python/demos/demo_cpu_inpainters.py +++ b/Wrappers/Python/demos/demo_cpu_inpainters.py @@ -167,7 +167,7 @@ pars = {'algorithm' : NVM_INP, \ } start_time = timeit.default_timer() -nvm_inp = NVM_INP(pars['input'], +(nvm_inp, mask_upd) = NVM_INP(pars['input'], pars['maskData'], pars['SW_increment'], pars['number_of_iterations']) diff --git a/Wrappers/Python/src/cpu_regularisers.pyx b/Wrappers/Python/src/cpu_regularisers.pyx index 19dd707..52befd7 100644 --- a/Wrappers/Python/src/cpu_regularisers.pyx +++ b/Wrappers/Python/src/cpu_regularisers.pyx @@ -397,4 +397,4 @@ def NVM_INP_2D(np.ndarray[np.float32_t, ndim=2, mode="c"] inputData, SW_increment, iterationsNumb, dims[0], dims[1], 1) - return outputData + return (outputData, maskData_upd) -- cgit v1.2.3