出力画像が正しく表示されない

Question

以下のコードにおけるimg_dstが思ったように表示されません。

import cv2
import math
import numpy as np
import random as rd

file_src = 'src.png'
file_dst = 'dst.png'

img_src = cv2.imread(file_src, 0) # 入力画像（グレースケール）の読み込み

cv2.namedWindow('src')
cv2.namedWindow('dst')


height, width = img_src.shape[:2]

img_dst = np.array([[0 for i in range(width)] for j in range(height)])

print(img_dst)

for y in range(height):
    for x in range(width):
        v = img_src[y][x]
        if v < rd.randrange(256):
            img_dst[y][x] = 0
        else:
            img_dst[y][x] = 255

print(img_dst)


cv2.imshow('src', img_src) # 入力画像を表示
cv2.imshow('dst', img_dst) # 出力画像を表示
cv2.imwrite(file_dst, img_dst) # 処理結果の保存
cv2.waitKey(0) # キー入力待ち
cv2.destroyAllWindows()

以下が出力です。

[[0 0 0 ... 0 0 0] [0 0 0 ... 0 0 0]
 [0 0 0 ... 0 0 0]
 ...
 [0 0 0 ... 0 0 0]
 [0 0 0 ... 0 0 0]
 [0 0 0 ... 0 0 0]]
[[255 255   0 ...   0   0   0]
 [  0 255   0 ...   0   0   0]
 [255   0 255 ...   0   0   0]
 ...
 [255   0   0 ...   0   0   0]
 [  0   0   0 ...   0   0 255]

このように、cv2.imshow('dst', img_dst) # 出力画像を表示 の直前でimg_dstは白い画素をもっているのに、実際にはこのプログラムを実行すると真っ黒の画像が表示されます。ただし、imwriteで処理結果を保存した先のファイルを開くと、ちゃんと白い画素を持つ、欲しかった画像が表示されます。
どうすれば解決しますか？

Answer 1

OpenCV: High-level GUI

If the image is 8-bit unsigned, it is displayed as is.

If the image is 16-bit unsigned or 32-bit integer, the pixels are divided by 256. That is, the value range [0,255*256] is mapped to [0,255].
If the image is 32-bit or 64-bit floating-point, the pixel values are multiplied by 255. That is, the value range [0,1] is mapped to [0,255].

とのことです。
img_dstのdtypeはint32になっていました。うまく表示されないのはそのためです。

img_dst = np.array([[0 for i in range(width)] for j in range(height)], dtype='uint8')

のようにdtypeにuint8を指定してあげることでうまく表示されるようになると思います。

Comments

[Sukabo]

うまくいきました。
ご回答ありがとうございました。