Matplotlib을 이용한 영상 출력

Matplotlib으로 컬러 영상 출력하기

• OpenCV에서 불러온 컬러 영상 데이터는 BGR 순서이지만, Matplotlib는 영상 데이터의 색상 순서가 RGB 순서라고 간주함
  → cv2.cvtColor() 함수 이용하여 색상 순서를 변경해야함

Matplotlib으로 그레이 스케일 영상 출력하기

• plt.imshow() 함수에서 컬러맵을 cmap='gray' 으로 지정해야함

import matplotlib.pyplot as plt
import cv2


# 컬러 영상 출력
imgBGR = cv2.imread('cat.bmp')
imgRGB = cv2.cvtColor(imgBGR, cv2.COLOR_BGR2RGB)

plt.axis('off')
plt.imshow(imgRGB)
plt.show()

# 그레이스케일 영상 출력
imgGray = cv2.imread('cat.bmp', cv2.IMREAD_GRAYSCALE)

plt.axis('off')
plt.imshow(imgGray, cmap='gray')
plt.show()

# 두 개의 영상을 함께 출력
plt.subplot(121), plt.axis('off'), plt.imshow(imgRGB)
plt.subplot(122), plt.axis('off'), plt.imshow(imgGray, cmap='gray')
plt.show()

Reference
1) 제로베이스 데이터스쿨 강의자료