・2 min read
AIにアミバとトキを学習させよう⑤＊オチあり
前回つくったモデルを読み込んで予測してみます。
使う木人形（デク）はこいつです。
さて結果は、、、
あれ、、、、？
ひ、暇ができたら学習データのクレンジングとか選定のやり直しでもっかいチャレンジしたいとおもいます。
うわらば、、、。
import keras
import sys, os
import numpy as np
from PIL import Image
from keras.models import load_model
imsize = (64, 64)
testpic = "./test.jpg"
keras_param = "./cnn.h5"
def load_image(path):
img = Image.open(path)
img = img.convert('RGB')
img = img.resize(imsize)
# 画像データをnumpy配列の形式に変更
img = np.asarray(img)
img = img / 255.0
return img
model = load_model(keras_param)
img = load_image(testpic)
prd = model.predict(np.array([img]))
print(prd)
prelabel = np.argmax(prd, axis=1)
if prelabel == 0:
print(">>> アミバ様")
elif prelabel == 1:
print(">>> トキ")
