diff --git a/enhance/ridge_filter.py b/enhance/ridge_filter.py index 9fd9e6a..302f586 100644 --- a/enhance/ridge_filter.py +++ b/enhance/ridge_filter.py @@ -77,7 +77,8 @@ def ridge_filter(im, orient, freq, kx, ky): sigmax = 1/unfreq[0]*kx; sigmay = 1/unfreq[0]*ky; - sze = np.round(3*np.max([sigmax,sigmay])); + #sze = np.round(3*np.max([sigmax,sigmay])); + sze = np.round(3*np.max([sigmax,sigmay])).astype('int'); x,y = np.meshgrid(np.linspace(-sze,sze,(2*sze + 1)),np.linspace(-sze,sze,(2*sze + 1))); @@ -138,4 +139,4 @@ def ridge_filter(im, orient, freq, kx, ky): newim[r][c] = np.sum(img_block * gabor_filter[int(orientindex[r][c]) - 1]); - return(newim); \ No newline at end of file + return(newim);