opencv のMatをbyte 列からつくりたい

Question

C言語で
Byte列をMATにするとき
BYTE * ptrImageData=入力
Mat newImg = Mat(nImageHeight, nImageWidth, CV_8UC3, ptrImageData);
みたいなことをしますが
これをPythonでするとき、どう書きますか？

Answer 1

OpenCV の Python 版では画像は numpy 配列として扱うので、numpy の frombuffer でバイト列を numpy 配列に変換できます。

numpy.frombuffer — NumPy v1.15 Manual

import numpy as np

data = b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\t\n\x0b\x0c\r\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a'

h, w, c = 3, 3, 3  # (画像の高さ, 画像の幅, チャンネル数)
img = np.frombuffer(data, dtype=np.uint8).reshape(h, w, c)
print(img)
# [[[ 0  1  2]
#   [ 3  4  5]
#   [ 6  7  8]]
#  [[ 9 10 11]
#   [12 13 14]
#   [15 16 17]]
#  [[18 19 20]
#   [21 22 23]
#   [24 25 26]]]

Comments

[kokawa2003]

うまく動作しました。ありがとうございました