Python Opencv中获取卷积核的实现代码

getStructuringElement(shape, ksize, anchor=None):

import os
import cv2
import numpy as np
def ErodeFilterRect(img_path='images/lenna.png'):
    img_src=cv2.imread(img_path)
    img=cv2.resize(src=img_src,dsize=(500,500))
    img=cv2.cvtColor(src=img,code=cv2.COLOR_BGR2GRAY)
    # kernel=np.ones((7,7),dtype=np.uint8)
    kernel=cv2.getStructuringElement(shape=cv2.MORPH_RECT,ksize=(3,3))
    print(kernel)
    img=cv2.erode(src=img,kernel=kernel,iterations=1)
    cv2.imshow('img_src',img_src)
    cv2.imshow('img',img)
    cv2.waiTKEy(0)
    cv2.destroyAllwindows()
def ErodeFilterELLIPSE(img_path='images/lenna.png'):
    img_src=cv2.imread(img_path)
    img=cv2.resize(src=img_src,dsize=(500,500))
    img=cv2.cvtColor(src=img,code=cv2.COLOR_BGR2GRAY)
    # kernel=np.ones((7,7),dtype=np.uint8)
    kernel=cv2.getStructuringElement(shape=cv2.MORPH_ELLIPSE,ksize=(3,3))
    print(kernel)
    img=cv2.erode(src=img,kernel=kernel,iterations=1)
    cv2.imshow('img_src',img_src)
    cv2.imshow('img',img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    
def ErodeFilterCROSS(img_path='images/lenna.png'):
    img_src=cv2.imread(img_path)
    img=cv2.resize(src=img_src,dsize=(500,500))
    img=cv2.cvtColor(src=img,code=cv2.COLOR_BGR2GRAY)
    # kernel=np.ones((7,7),dtype=np.uint8)
    kernel=cv2.getStructuringElement(shape=cv2.MORPH_CROSS,ksize=(3,3))
    print(kernel)
    img=cv2.erode(src=img,kernel=kernel,iterations=1)
    cv2.imshow('img_src',img_src)
    cv2.imshow('img',img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
if __name__ == '__main__':
    print('PyCharm')
    # ErodeFilterRect()
    # ErodeFilterELLIPSE()
    ErodeFilterCROSS()