Advertisement
Guest User

Untitled

a guest
Mar 31st, 2020
88
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.89 KB | None | 0 0
  1. def canny(iImage, thr_type = 1):
  2.     iImage = np.asarray(iImage)
  3.     iImageG = rv.colorToGray(iImage)
  4.     # glajenje slike
  5.     iKernel = np.ones([5,5])/25
  6.     iImageGL = ndimage.convolve(iImageG, iKernel)
  7.     # dolocimo kriterij za povezovalne tocke
  8.     if thr_type == 1:
  9.         m = np.median(iImageGL)
  10.         std = np.std(iImageGL)
  11.     #elif thr_type == 2:
  12.     #    m = np.median(iImageGL)
  13.     #    std = np.std(iImageGL)
  14.     y, x = iImageGL.shape
  15.     grad_0 = grad_45 = grad_90 = grad_135 = np.zeros_like(iImageGL)
  16.     for dy in range(1, y-1):
  17.         for dx in range(1, x-1):
  18.             grad_0[dy, dx] = abs(iImageGL[dy, dx+1]-iImageGL[dy, dx-1])
  19.             grad_45[dy, dx] = abs(iImageGL[dy+1, dx]-iImageGL[dy-1, dx])
  20.             grad_90[dy, dx] = abs(iImageGL[dy+1, dx]-iImageGL[dy-1, dx])
  21.             grad_135[dy, dx] = abs(iImageGL[dy+1, dx]-iImageGL[dy-1, dx])
  22.     return None
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement