canny 边缘检测（小作业）

作业需要 ，要的自取

import cv2
import numpy as np
import math
import matplotlib.pyplot as plt

img=cv2.imread("cat.jpg")

#灰度图
gray_img= cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

#1. 高斯滤波
#(5, 5)表示高斯矩阵的长与宽都是5，标准差取1
gray_img=cv2.GaussianBlur(gray_img,(5,5),1)
gaussian=np.uint8(np.copy(gray_img))

#展示一下高斯后结果
plt.imshow(gaussian)
plt.show()

 

#2.sobel 计算梯度幅值
W,H=gray_img.shape[:2]  #应该就是0-2 省略了0 然后分别赋值
#w==x h==y

dx=np.zeros((W,H),dtype=np.float)
dy=np.zeros((W,H),dtype=np.float)
d=np.zeros((W,H),dtype=np.float)
angle=np.zeros((W,H),dtype=np.float)

Xsobel=np.array([[1,2,1],
                 [0,0,0],
                 [-1,-2,-1]])
Ysobel=np.array([[1,0,-1],
                [2,0,-2],
                [1,0,-1]])

#需要最外边一圈为0填充