python opencv sobel算法

import cv2
import numpy as np
import tkinter

from tkinter import  filedialog

selectFileName = tkinter.filedialog.askopenfilename(title='选择文件')  # 选择文件

img = cv2.imread(selectFileName, 0)
sobel_x=[[-1,0,1],[-2,0,2],[-1,0,1]]#水平模板
sobel_y=[[-1,-2,-1],[0,0,0],[1,2,1]]#垂直模板

src=img.shape
high=src[0]
width=src[1]
res=np.zeros((high,width),np.uint8)
for i in range(1,high-1):
    for j in range(1,width-1):
        imgg=[[img[i-1,j-1],img[i-1,j],img[i-1,j+1]],[img[i,j-1],img[i,j],img[i,j+1]],[img[i+1,j-1],img[i+1,j],img[i+1,j+1]]]
        imggjuzheng=np.array(imgg)
        roberts_x = np.array(sobel_x)
        roberts_y = np.array(sobel_y)
        var_x=np.sum(roberts_x*imggjuzheng)#将所有元素相加
        var_y=np.sum(roberts_y*imggjuzheng)
        res[i,j]=np.abs(var_x)+np.abs(var_y)
cv2.imshow("ima",res)
cv2.waitKey(0)
cv2.destroyAllWindows()