(Python) array(image) to base64, base64 to image

2차원 배열을 이미지로 바꾼 후 base64 형태로 바꾸는 것이라서, 이미지 to base64도 가능합니다.

import base64
from PIL import Image
from io import BytesIO

#array_img는 2차원배열로 된 image임 즉, 1024*1024인 경우는 행이 1024개 열이 1024개인 2차원배열이다

rawBytes = BytesIO()

#2차원 배열의 type은 uint8이여야 인코딩 가능
img_buffer = Image.fromarray(array_img.astype('uint8'))

img_buffer.save(rawBytes, 'PNG')
rawBytes.seek(0)
base64_img = base64.b64encode(rawBytes.read())

 

반대의 경우는 아래와 같이 하면 됩니다.

img = Image.open(BytesIO(base64.b64decode(base64_img)))