我正在尝试从模型中加载节省的重量,以便可以使用它进行预测,但是我一直收到此错误 AttributeError:“ NoneType”对象没有属性“ predit_classes”
我知道我尝试过的代码主要用于这样使用的模型: 模型= Sequential(),但我不知道该怎么做。
test_model=model.load_weights('second_try.h5')
img = load_img('0008_00_00_01_219.jpg',False,target_size=(150,150))
x=img_to_array(img)
x = np.expand_dims(x, axis=0)
img_class = test_model.predict_classes(x)
print ("Class:", img_class)
答案 0 :(得分:0)
load_weights
不返回模型(不返回任何内容),model
变量中已经有一个模型,因此您只需要执行以下操作:
model.load_weights('second_try.h5')
img = load_img('0008_00_00_01_219.jpg',False,target_size=(150,150))
x=img_to_array(img)
x = np.expand_dims(x, axis=0)
img_class = model.predict_classes(x)
print ("Class:", img_class)