Python

numpy array 구조를 image로 저장하기

언킴 2022. 6. 1. 22:04
반응형

이미지를 호출하기 위해서는 다음과 같은 패키지가 필요하다. 

from PIL import Image 
import numpy as np

이미지를 호출하고, 이를 np.array로 변환하면 28*28 이미지는 28*28*1 차원을 가지는 array가 될 것이다. 마지막 1은 channel을 의미하며, channel이 3인 경우 RGB 컬러 이미지를 의미한다. 

IMG_DIR = f'../datasets/images'
img_path = os.path.join(IMG_DIR, file_name + '.jpg')

img = np.array(Image.open(img_path))

위 코드는 이미지가 저장된 경로에 접근해 jpg 파일을 호출하는 코드다. 호출된 코드는 아래와 같은 구조를 가지고 있다. 

array 구조를 이미지로 저장하고 싶을 땐 Image.fromarray 함수를 사용하면 쉽게 저장할 수 있다.

Image.fromarray(transforms_dict['image'])