NumpyとOpenCVを用いて単色画像を生成したい

Question

前提・実現したいこと

NumpyとOpencvを用いて単色画像を生成したい。

発生している問題・エラーメッセージ

コードのようにRGB値を指定してもどちらも全面黒の画像が表示される。

該当のソースコード

import numpy as np
import cv2

img_size = (512, 512)

img = np.array([[[0, 0, 0] for _ in range(img_size[1])] for _ in range(img_size[0])])
img1 = np.array([[[255, 255, 255] for _ in range(img_size[1])] for _ in range(img_size[0])])

cv2.imshow("sample", img)
cv2.imshow("sample1", img1)
cv2.waitKey(0)
cv2.destroyAllWindows()

試したこと

matplotlibを用いて表示したところ指定したRGB値が反映された。

補足情報（FW/ツールのバージョンなど）

Anaconda3を用いており、パッケージはAnaconda Navigatorを用いてインストールした。
よろしくお願いします（・o・）

Answer 1

OpenCV で画像を扱うには ndarray の型 dtype は np.uint8 である必要があります。以下のように明示的に指定する必要があります。
また、すべての同じ値の配列を作成する場合は、ループではなく、numpy.full() 関数をお使いください。

import numpy as np
import cv2

# BGR = (255, 255, 255) の大きさが (512, 512) の画像を作成する。
img1 = np.full((512, 512, 3), 255, dtype=np.uint8)

# BGR = (255, 0, 0) の大きさが (512, 512) の画像を作成する。
img2 = np.full((512, 512, 3), (255, 0, 0), dtype=np.uint8)

cv2.imshow("img1", img1)
cv2.imshow("img2", img2)
cv2.waitKey(0)
cv2.destroyAllWindows()

Comments

[Pe__ta___]

助かりました、ありがとうございます。