1. 程式人生 > >keras儲存的模型用來做預測

keras儲存的模型用來做預測

from keras.models import load_model
from keras.preprocessing import image
import numpy as np

file_path='4d.jpg' ##隨便啥圖都可以.
img=image.load_img(file_path,target_size=(32,32))
x=image.img_to_array(img)
x=np.expand_dims(x,axis=0)


##儲存權重model.save_weights(),儲存模型model.save()
model=load_model('model_weight.h5')

y=model.predict(x)

print(y)