candy算子python_实现Canny算法 python

1 #writer：wojianxinygcl@163.com

2 #date ：2020.3.20

3 importcv24 importnumpy as np5 importmatplotlib.pyplot as plt6

7 defCanny(img):8

9 #Gray scale

10 defBGR2GRAY(img):11 b =img[:, :, 0].copy()12 g = img[:, :, 1].copy()13 r = img[:, :, 2].copy()14

15 #Gray scale

16 out = 0.2126 * r + 0.7152 * g + 0.0722 *b17 out =out.astype(np.uint8)18

19 returnout20

21

22 #Gaussian filter for grayscale

23 def gaussian_filter(img, K_size=3, sigma=1.4):24

25 if len(img.shape) == 3:26 H, W, C =img.shape27 gray =False28 else:29 img = np.expand_dims(img, axis=-1)30 H, W, C =img.shape31 gray =True32

33 ## Zero padding

34 pad = K_size // 2

35 out = np.zeros([H + pad * 2, W + pad * 2, C], dtype=np.float)36 out[pad : pad + H, pad : pad + W] =img.copy().astype(np.float)37

38 ## prepare Kernel

39 K = np.zeros((K_size, K_size), dtype=np.float)40 for x in range(-pad, -pad +K_size):41 for y in range(-pad, -pad +K_size):42 K[y + pad, x + pad] = np.exp( - (x ** 2 + y ** 2) / (2 * sigma *sigma))43 #K /= (sigma * np.sqrt(2 * np.pi))

44 K /= (2 * np.pi * sigma *sigma)45 K /=K.sum()46

47 tmp =out.copy()48

<