Computer Vision/opencv(python)

[03]Matplotlib을 이용한 영상 출력

jwjwvison 2021. 3. 11. 23:18
  • 컬러 영상 출력
    • 컬러 영상의 색상 정보가  RGB 순이어야함
    • cv2.imread() 함수로 불러온 영상의 색상 정보는 BGR 순서이므로 이를 RGB 순서로 변경해야 함 --> cv2.cvtColor() 함수 이용
  • 그레이스케일 영상 출력
    • plt.imshow() 함수에서 컬러맵을 cmap='gray'으로 지정
  • Matplotlib을 이용한 영상 출력
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)   #반드시 RGB순서로 
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)    #몇개의 행(1), 몇개의 열(2), 몇번째 열에 그림을 그릴것이냐(1,2)
plt.subplot(122), plt.axis('off'), plt.imshow(imgGray, cmap='gray')
plt.show()