diff --git a/skimage/restoration/_denoise_cy.pyx b/skimage/restoration/_denoise_cy.pyx index b679b488..3c71b2e4 100644 --- a/skimage/restoration/_denoise_cy.pyx +++ b/skimage/restoration/_denoise_cy.pyx @@ -6,7 +6,6 @@ cimport numpy as cnp import numpy as np from libc.math cimport exp, fabs, sqrt -from libc.stdlib cimport malloc, free from libc.float cimport DBL_MAX from .._shared.interpolation cimport get_pixel3d from ..util import img_as_float @@ -16,10 +15,10 @@ cdef inline double _gaussian_weight(double sigma, double value): return exp(-0.5 * (value / sigma)**2) -cdef double* _compute_color_lut(Py_ssize_t bins, double sigma, double max_value): +cdef double[:] _compute_color_lut(Py_ssize_t bins, double sigma, double max_value): cdef: - double* color_lut = malloc(bins * sizeof(double)) + double[:] color_lut = np.empty(bins, dtype=np.double) Py_ssize_t b for b in range(bins): @@ -28,10 +27,10 @@ cdef double* _compute_color_lut(Py_ssize_t bins, double sigma, double max_value) return color_lut -cdef double* _compute_range_lut(Py_ssize_t win_size, double sigma): +cdef double[:] _compute_range_lut(Py_ssize_t win_size, double sigma): cdef: - double* range_lut = malloc(win_size**2 * sizeof(double)) + double[:] range_lut = np.empty(win_size**2, dtype=np.double) Py_ssize_t kr, kc Py_ssize_t window_ext = (win_size - 1) / 2 double dist @@ -74,16 +73,16 @@ def _denoise_bilateral(image, Py_ssize_t win_size, sigma_range, double[:, :, ::1] cimage double[:, :, ::1] out - double* color_lut - double* range_lut + double[:] color_lut + double[:] range_lut Py_ssize_t r, c, d, wr, wc, kr, kc, rr, cc, pixel_addr, color_lut_bin double value, weight, dist, total_weight, csigma_range, color_weight, \ range_weight double dist_scale - double* values - double* centres - double* total_values + double[:] values + double[:] centres + double[:] total_values if sigma_range is None: csigma_range = image.std() @@ -95,20 +94,20 @@ def _denoise_bilateral(image, Py_ssize_t win_size, sigma_range, if max_value == 0.0: raise ValueError("The maximum value found in the image was 0.") + if mode not in ('constant', 'wrap', 'symmetric', 'reflect', 'edge'): + raise ValueError("Invalid mode specified. Please use `constant`, " + "`edge`, `wrap`, `symmetric` or `reflect`.") + cdef char cmode = ord(mode[0].upper()) + cimage = np.ascontiguousarray(image) out = np.zeros((rows, cols, dims), dtype=np.double) color_lut = _compute_color_lut(bins, csigma_range, max_value) range_lut = _compute_range_lut(win_size, sigma_spatial) dist_scale = bins / dims / max_value - values = malloc(dims * sizeof(double)) - centres = malloc(dims * sizeof(double)) - total_values = malloc(dims * sizeof(double)) - - if mode not in ('constant', 'wrap', 'symmetric', 'reflect', 'edge'): - raise ValueError("Invalid mode specified. Please use `constant`, " - "`edge`, `wrap`, `symmetric` or `reflect`.") - cdef char cmode = ord(mode[0].upper()) + values = np.empty(dims, dtype=np.double) + centres = np.empty(dims, dtype=np.double) + total_values = np.empty(dims, dtype=np.double) for r in range(rows): for c in range(cols): @@ -146,12 +145,6 @@ def _denoise_bilateral(image, Py_ssize_t win_size, sigma_range, for d in range(dims): out[r, c, d] = total_values[d] / total_weight - free(color_lut) - free(range_lut) - free(values) - free(centres) - free(total_values) - return np.squeeze(np.asarray(out))