sklearn의 datasets에 있는 faces데이터로부터 이미지를 띄워보자
- 데이터셋 받아오기
import numpy as np from sklearn import datasets faces=datasets.fetch_olivetti_faces()
- 데이터 하나의 크기 확인 (1, 4096) 크기의 1차원 배열
faces.data[0].shape
- 이미지 하나의 크기 확인 (64, 64) 크기의 2차원 배열
faces.images[0].shape
- 이미지 띄우기
from matplotlib import pyplot as plt plt.imshow(faces.images[0])
4-2. 이미지 회색으로 띄우기
plt.imshow(faces.images[0],cmap=plt.cm.gray)
- data[0]과 images[0]은 사실 같은 데이터인데 형태만 다름
그래서 아래 코드는 오류가 남plt.imshow(faces.data[0],cmap=plt.cm.gray)
- 다음과 같이 image데이터 크기 (64,64)로 맞춰줘야함
plt.imshow(faces.data[0].reshape(64,64), cmap=plt.cm.gray)