머신러닝_티쳐블머신 모델 사용하기

티쳐블머신 모델 사용하기

• pillow 설치하기

  !pip install pillow

• import하기

  from tensorflow.keras.models import load_model
   from PIL import Image, ImageOps
   import numpy as np

• 실행 소스

 # 학습된 모델로딩
 model = load_model('animal_keras_model.h5')
 #예측시킬 이미지 데이터를 넣을 변수
 data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32) 
 #예측시킬 이미지 로딩
 image = Image.open('./predict1.jpg')
 #똑같은 이미지 크기로 변경
 size = (224, 224)
 image = ImageOps.fit(image, size, Image.ANTIALIAS)

 #numpy 타입으로 변경
 image_array = np.asarray(image)

 normalized_image_array = (image_array.astype(np.float32) / 127.0) - 1
 # 불러운 numpy 타입의 이미지를 변수에 대입
 data[0] = normalized_image_array

 # 예측
 prediction = model.predict(data)
 print(prediction)