创建我的自定义内核垫

Question

如何在 OpenCV Python 中创建自己的内核？

我正在尝试在下面创建一个 sobel 内核，但出现错误：

AttributeError: 'module' 对象没有属性 'mat'

def sobel(src):

    kernelX = [
        [ -1, 0, 1 ],
        [ -2, 0, 2 ],
        [ -1, 0, 1 ]
    ]

    kernelY = [
        [ -1, -2, -1 ],
        [  0,  0,  0 ],
        [  1,  2,  1 ]
    ]

    kernelXMat = cv2.mat(3, 3, cv2.cv2.CV_32FC1, kernelX)
    kernelYMat = cv2.mat(3, 3, cv2.CV_32FC1, kernelY)

    horizGradient = cv2.filter2D(src, -1, kernelXMat)
    vertGradient = cv2.filter2D(src, -1, kernelYMat)

    return [horizGradient, vertGradient]

Answer 1

OpenCV Mat 的 Python 等效项是 numpy.array。可以在 python 中按如下方式创建指定的内核。

import numpy as np

kernelXMat = np.array(kernelX, dtype=float);
kernelYMat = np.array(kernelY, dtype=float);