画像ファイルをbase64形式に変換する方法

Question

前提・実現したいこと

Python(Flask)を使って画像を処理するAPIを作りたい。
画像を保存する処理が入ってしまうと、使わない画像が溜まってしまうため、cv2.imwriteを使わずに、画像ファイルをpostする（base64形式に変換する）方法を探している。

こちらの記事を参考にコードを作成したのですが、画像ファイルをbase64に変換する前のimwriteやopenを使わずにbase64形式に変換する方法が、そもそもそういったことが可能かどうかも含めわかりません。

該当のソースコード

from flask import *
import json
import io
from flask_cors import CORS
import numpy as np
import cv2
import base64

@app.route('/test', methods=["POST"])
def post():
    print("POSTED")
    stream = request.files["image"].stream
    img_array = np.asarray(bytearray(stream.read()), dtype=np.uint8)
    img = cv2.imdecode(img_array, 0)
    cv2.imwrite('result.png', img)
    with open('result.png', "rb") as f:
        img_base64 = base64.b64encode(f.read()).decode('utf-8')

    response = make_response(img_base64)
    response.headers.set('Content-Type', request.content_type)

    return response

試したこと

画像をdecode, encodeする際にそれぞれどの様なデータになっているのか、そもそもなぜ画像をPOSTする際にbase64などに変換しないといけないのかなど調べているのですが、なかなかこれという答えに辿り着けずにいます。

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

Flask 1.1.2

Answer 1

以下の手順で ndarray 型の画像データを直接 base64 に変換できます。

1. cv2.imencode() で画像データをバイナリデータにする
2. バイナリデータを binascii.b2a_base64() で bytes 列にする
3. (str 型でほしい場合) bytes 列を decode('ascii') で str にする

import cv2
import binascii

img = cv2.imread("sample.jpg")

# ndarray -> binary
ret, data = cv2.imencode(".jpg", img, [cv2.IMWRITE_JPEG_QUALITY, 90])
# binary -> base64 ascii
base64_ascii = binascii.b2a_base64(data).decode('ascii')

base64 に変換できていることを確認したコード

Jupyter Notebook 上にインライン画像として img タグ表示して確認

import cv2
import binascii
from IPython.display import HTML, display

# jpg の場合
img = cv2.imread("sample.jpg")
ret, jpg_data = cv2.imencode(".jpg", img, [cv2.IMWRITE_JPEG_QUALITY, 90])
base64_ascii = binascii.b2a_base64(jpg_data).decode("ascii")

img_tag = f'<img src="data:image/jpeg;base64,{base64_ascii}">'
display(HTML(img_tag))

# png の場合
img = cv2.imread("bird.png")
ret, png_data = cv2.imencode(".png", img)
base64_ascii = binascii.b2a_base64(png_data).decode("ascii")

img_tag = f'<img src="data:image/png;base64,{base64_ascii}">'
display(HTML(img_tag))

Comments

[E28]

ありがとうございます！
いただいた回答を参考に知識を深めて行きたいと思います！