如何在 Python 中输入提示 OpenCV 图像？

Question

我知道在Python中OpenCV图像是numpy数组，对应于c++中的cv::Mat。

这个问题是关于在 python 函数中放入什么类型提示来正确限制 OpenCV 图像（甚至可能对于特定类型的 OpenCV 图像）。

我现在做的是：

import numpy as np
import cv2

Mat = np.ndarray

def my_fun(image: Mat):
    cv2.imshow('display', image)
    cv2.waitKey()

有没有更好的方法在 python 中为 OpenCV 图像添加输入信息？

Answer 1

您可以将其指定为numpy.typing.NDArray条目类型。例如，

import numpy as np

Mat = np.typing.NDArray[np.uint8]

def my_fun(img: Mat):
    pass