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()