[matplotlib] matplotlib.pyplot을 이용한 이미지 시각화 총정리

0Install

pip install matplotlib

 

 

Import

import matplotlib.pyplot as plt

 

* 버전 확인

import matplotlib
print(matplotlib.__version__)

 

 

plot image

#load image
import cv2

img = cv2.imread('dog.jpg')
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

#show iage
plt.imshow(img)

 

 

figure 크기 조절

plt.figure(figsize=(15, 15))
plt.imshow(img)

 

 

colorbar 표시

plt.imshow(img[...,0]) #마지막 채널만 표시
plt.colorbar()

 

* Colorbar 위치 이동 (ver 3.4부터 가능)

plt.imshow(img[...,0])
plt.colorbar(location='left') #left, right(default), top, bottom

 

 

최솟값, 최댓값 설정

plt.imshow(img[...,0], vmin=50, vmax=100)
plt.colorbar()

 

 

Grayscale

plt.imshow(img[...,0], cmap='gray')

 

 

Set title

plt.imshow(img)
plt.title('dog')

 

 

x 축 label, y 축 label 설정

plt.imshow(img)
plt.xlabel('x-axis')
plt.ylabel('y-axis')

 

 

눈금 없애기

plt.imshow(img)
plt.axis('off')

 

 

Subplot (이미지 여러개 띄우기)

img2 = cv2.imread('cat.jpg')
img2 = cv2.cvtColor(img2, cv2.COLOR_BGR2RGB)

plt.figure(figsize=(10, 4))

plt.subplot(1, 2, 1)
plt.imshow(img)

plt.subplot(1, 2, 2)
plt.imshow(img2)

 

* Subplot title & 전체 title (suptitle)

plt.figure(figsize=(10, 4))

plt.subplot(1, 2, 1)
plt.imshow(img)
plt.title('dog')

plt.subplot(1, 2, 2)
plt.imshow(img2)
plt.title('cat')

plt.suptitle('animals')