![見出し画像](https://assets.st-note.com/production/uploads/images/100312748/rectangle_large_type_2_a61e94ea19baf715db37add3b0e7c97d.png?width=1200)
OpenCVのimencodeとimdecodeでのbytesのやりとり
画像をimreadで読み込んだ後に、imencodeで取得したbytesを再度imdecodeで画像データに戻すメモです。
画像の読み込みと一部切り取り
ついでに切り取りもしちゃってます。
import cv2
fp = os.path.join(PATH_TO_IMAGE_DIR, IMAGE_NAME)
image = cv2.imread(fp)
crop = image[100:200, 100:200]
print(type(crop))
# <class 'numpy.ndarray'>
ちなみにGoogle Colabだと以下で画像の確認ができます。
from google.colab.patches import cv2_imshow
cv2_imshow(crop)
imencodeでbytesを取得
今回読み込んだ画像はpngでしたので。
retval, buf = cv2.imencode('.png', crop)
# retval -> True
# buf -> array([137, 80, 78, ..., 66, 96, 130], dtype=uint8)
crop_bytes = buf.tobytes()
print(type(buf))
# <class 'numpy.ndarray'>
print(type(crop_bytes))
# <class 'bytes'>
imdecodeで再度画像データに戻す
imdecodeにはbytesは渡せないので、一度ndarrayに変換してあげる必要があります。
crop_ndarray = np.frombuffer(crop_bytes, np.uint8)
print(type(crop_ndarray))
# <class 'numpy.ndarray'>
decoded = cv2.imdecode(crop_ndarray, cv2.IMREAD_UNCHANGED)
print(type(decoded))
# <class 'numpy.ndarray'>
cv2_imshow(decoded)
この記事が気に入ったらサポートをしてみませんか?