Matplotlib 이미지 파일 읽기/저장하기

---

안녕하세요~ 꽁냥이에요. 이번 포스팅에서는 Matplotlib을 이용하여 이미지를 읽고 저장하는 방법에 대해서 알아보겠습니다.

 

1. 이미지 파일 읽기

2. 이미지 저장하기

---

   1. 이미지 파일 읽기

먼저 필요한 모듈을 임포트 합니다. 이때 Matplotlib에서 이미지 파일을 읽어 오기 위해서 matplotlib.image 모듈이 필요합니다.

 

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as img

 

이제 이미지 파일을 읽어볼게요. 아래 코드는 sample.jpg 이미지 파일을 읽어오는 코드예요.

 

image_path = './sample.jpg'
image = img.imread(image_path) ## RGB 값
plt.imshow(image) ## 이미지 출력
plt.axis('off') ## 축 제거
plt.show()

 

line 2

이미지를 바로 읽는 것은 아니고 먼저 이미지 파일을 RGB 값으로 변환해야 합니다.

 

line 3

실제로 이미지를 출력하는 부분입니다. Matplotlib에서는 imshow를 이용하여 이미지를 출력할 수 있습니다.

 

---

   2. 이미지 저장하기

이번에는 그래프를 이미지로 저장하는 방법에 대해서 알아보겠습니다. 어렵지 않아요. Matplotlib에서 제공하는 savefig 함수를 이용하면 됩니다. savefig에서 dpi 인자를 통해 해상도를 조절할 수 있어요. 

 

fig = plt.figure(figsize=(8,8))
x = np.random.rand(10)
y = np.random.rand(10)
plt.plot(x,y) ## 선 그래프
plt.savefig(f'./line_plot.jpg', dpi=300)

 

---

이미지를 처리하는 모듈은 여러 가지(opencv, PIL)가 있는데요. 이번 포스팅을 통해서 Matplotlib으로도 이미지 파일을 읽고 출력하는 걸 알려드리고 싶었어요 ㅎㅎ. 알아두시면 언젠간 쓸모가 있을 거예요~~ 

 

---