NumPy配列をcv::UMat型に変換したい。

Question

実現したいこと

NumPy配列をcv::UMat型に変換したい。

状況

Base64でエンコードしたJpeg画像をPythonを用いてBase64でデコードした画像を保存する機能を作っています。cv2.cvtColorの第一引数にNumpy配列を渡していますがcv::UMat型である必要があるとエラーが発生しました。Numpy配列をcv::UMat型に変換する方法が思いつかず困っています。

該当のソースコード・エラーメッセージ

#バイナリデータ <- base64でエンコードされたデータ  
img_binary = base64.b64decode(img_base64)
#バイナリーストリーム <- バリナリデータ
img_binarystream = io.BytesIO(img_binary)
#PILイメージ <- バイナリーストリーム
img_pil = Image.open(img_binarystream)
#numpy配列(RGBA) <- PILイメージ
img_numpy = np.asarray(img_pil)

#numpy配列(BGR) <- numpy配列(RGBA)
img_numpy_bgr = cv2.cvtColor(img_numpy, cv2.COLOR_RGBA2BGR)

img_numpy_bgr = cv2.cvtColor(img_numpy, cv2.COLOR_RGBA2BGR)
TypeError: Expected cv::UMat for argument 'src'

ソースコード

import cv2
import base64
import numpy as np
import io
from PIL import Image

#エンコードされたJpeg画像
target_file=r"D:\Images\replace.txt"

#画像保存先
image_file=r"D:\Images\im.jpg"

with open(target_file, 'rb') as f:
    img_base64 = f.read()

#バイナリデータ <- base64でエンコードされたデータ  
img_binary = base64.b64decode(img_base64)

#バイナリーストリーム <- バリナリデータ
img_binarystream = io.BytesIO(img_binary)

#PILイメージ <- バイナリーストリーム
img_pil = Image.open(img_binarystream)

#numpy配列(RGBA) <- PILイメージ
img_numpy = np.asarray(img_pil)

#numpy配列(BGR) <- numpy配列(RGBA)
img_numpy_bgr = cv2.cvtColor(img_numpy, cv2.COLOR_RGBA2BGR)

#表示確認
#cv2.imshow('window title', img_numpy_bgr)
#cv2.waitKey(0)
#cv2.destroyAllWindows()

#画像を保存する場合
cv2.imwrite(image_file,img_numpy_bgr)

試したこと

最初の一回目は保存に成功しましたが、それ以降なぜかエラーが発生するようになりました。
UMatに関して調べてみましたがOpenCVに疎く、理解できずこの度質問させていただきました。
よろしくお願い致します。

Comments

[pepperleaf]

一回目と二回目以降で何か違うことはありませんか?
あと、replace_fileが未定義みたいですが、、。

[TsubasaSato]

ご指摘ありがとうございます。
投稿時に間違ってreplace_fileと書いてしまったようです。修正しました。

1回目はファイルのパスに日本語が入っていた事ですかね、、その後エラーが発生したためファイル名を英語にする事でエラーを発生させず実行できるかと思い、パスを英語に変えましたがエラーを止めることはできませんでした。

[pepperleaf]

自分で確認したわけではありませんが、ファイル名、パス名に漢字が入ると Pythonは、ダメと聞きますが、ちょっとちょっと違うでしょうか。
UMATは、調べた限りでは、OpenCVの新しい(と言ってもかなり前の)画像フォーマットですね。関数の使い方そのものは問題なさそう。
調べてみたいですが、間に合っていません。詳しい人、いないでしょうか?

Answer 1

画像 <-> base64 テキストの変換は以下のようすると、できます。

画像 -> base64 テキスト

import cv2
import base64

img = cv2.imread('test.png')

# raw image -> jpg
ret, jpg = cv2.imencode('.jpg', img)

with open('test.txt', 'wb') as f:
    # jpg -> base64
    encoded = base64.b64encode(jpg.tostring())
    f.write(encoded)

base64 テキスト -> 画像

with open('test.txt', 'rb') as f:
    img_base64 = f.read()
# base64 -> jpg
jpg = base64.b64decode(img_base64)
jpg = np.frombuffer(jpg, dtype=np.uint8)
# jpg -> raw image
img = cv2.imdecode(jpg, cv2.IMREAD_COLOR)
cv2.imwrite('decoded.png', img)

Comments

[TsubasaSato]

ありがとうございました。
頂きました回答で無事にデコードすることができました！

他の方も回答していただきましてありがとうございました。