wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-03 10:36:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

FIX: Move a argument check to avoid leaking malloc'd arrays.

Browse Source
This commit is contained in:
John Wiggins
2015-12-26 10:55:28 -06:00
parent f3392bc4f5
commit e528d47d5b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/_denoise_cy.pyx
+5 -5
View File
@@ -95,6 +95,11 @@ 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)
@@ -105,11 +110,6 @@ def _denoise_bilateral(image, Py_ssize_t win_size, sigma_range,
centres = <double*>malloc(dims * sizeof(double))
total_values = <double*>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())
for r in range(rows):
for c in range(cols):
total_weight = 0