pyplotで作成したグラフデータを保存せずに画像データとして読み込みたい

Question

#やりたいこと
ヒストグラムの色データを1枚の画像データにして返したいが、
画像は保存したくない
#作成中のプログラム

import cv2,os
import matplotlib.pyplot as plt
img = cv2.imread("1.jpeg")
hist_blue = cv2.calcHist(img,[0],None,[256],[0,256])
hist_green = cv2.calcHist(img,[1],None,[256],[0,256])
hist_red = cv2.calcHist(img,[2],None,[256],[0,256])
# pltデータ作成
fig, ax = plt.subplots(figsize=(10, 5))
ax.set_title("blue")
ax.plot(hist_blue,color="blue")
fig, ax = plt.subplots(figsize=(10, 5))
ax.set_title("green")
ax.plot(hist_green,color="green")
fig, ax = plt.subplots(figsize=(10, 5))
ax.set_title("red")
ax.plot(hist_red,color="red")
#画像を見る
plt.show()
#保存
plt.savefig("foo.png")
hist_img = cv2.imread("foo.png")
#削除
os.remove("foo.png")

とまぁ、こんな感じでRGBのヒストグラムが作成されるわけですが、
保存して削除という行為を省くことは可能ですか?

Answer 1

以下のように、セーブとリードのところを、実ファイルではなくバイトストリーム使うと、質問者様のやりたいことは可能です。単純に行かなくて、少しおまじないが必要のようです。

参考: opencvでBytesIOイメージをロードする

元のコード

plt.savefig("foo.png")
hist_img = cv2.imread("foo.png")

修正後コード

import io
import numpy as np
with io.BytesIO() as bs:
    plt.savefig(bs, format='png')
    bs.seek(0)
    file_bytes = np.asarray(bytearray(bs.read()), dtype=np.uint8)
    hist_img = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)

Comments

[sk-sora--ypi]

ありがとうございます、
バイトで扱うことで可能になるんですね。