python - 使用 cv2 在 python 中创建多 channel 零垫

Question

我想用 cv2 opencv 包装器在 python 中创建一个多 channel mat 对象。

我在网上找到了将 c++ Mat::zeros 替换为 numpy.zeros 的示例，这看起来不错。但似乎没有适合的多 channel 类型..

看代码:

import cv2
import numpy as np

size = 200, 200
m = np.zeros(size, dtype=np.uint8) # ?
m = cv2.cvtColor(m, cv2.COLOR_GRAY2BGR)
p1 = (0,0)
p2 = (200, 200)
cv2.line(m, p1, p2, (0, 0, 255), 10)

cv2.namedWindow("draw", cv2.CV_WINDOW_AUTOSIZE)
while True:
    cv2.imshow("draw", m)

    ch = 0xFF & cv2.waitKey(1)
    if ch == 27:
        break
cv2.destroyAllWindows()

我想避免 m = cv2.cvtColor(m, cv2.COLOR_GRAY2BGR) 但 cv2.CV_8UC3 np.uin32 都不起作用。

一些提示？

Answer 1

试试这个作为 size:

size = 200, 200, 3
m = np.zeros(size, dtype=np.uint8)

基本上，我为找到矩阵所需的参数所做的工作是:

img = cv2.imread('/tmp/1.jpg')
print img.shape, img.dtype
# (398, 454, 3), uint8

但也可以在 OpenCV 文档中找到它。