keras保存的模型用来做预测

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/xfjjs_net/article/details/84630023
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)

猜你喜欢

转载自blog.csdn.net/xfjjs_net/article/details/84630023