【发布时间】:2021-05-26 19:15:16
【问题描述】:
我正在尝试使用 Keras 的 VGG16,我标记了 include_top=false
但我遇到了ValueError: Attempt to convert a value (None) with an unsupported type (<class 'NoneType'>) to a Tensor.
这是代码:
input_shape = (150,150,3)
model_1 = VGG16(weights='imagenet',include_top=False,input_shape=input_shape)
Last_layer=model_1.layers[-1].output
print(Last_layer)
print(type(Last_layer))
Model_Vgg=keras.layers.Flatten()(Last_layer) #<---- error rised here
#Model_Vgg=keras.Model(model.input,layer_output)
Model_Vgg = layers.Dropout(0.5)(Model_Vgg)
Model_Vgg = layers.Dense(units=3, activation='softmax') (Model_Vgg)
model = keras.Model(inputs =model_1.input,outputs = Model_Vgg )
model.compile(loss='categorical_crossentropy',optimizer=optimizers.SGD(lr=0.005708),metrics=['accuracy'])
monitor = EarlyStopping(monitor='accuracy',patience=50, mode='auto', restore_best_weights=True)
model.fit(X_Train,Y_Train,callbacks=[monitor],epochs=280,verbose=0)
(loss, accuracy) = model.evaluate(X_Test, Y_Test, batch_size=32, verbose=50)
print("[INFO] loss={:.4f}, accuracy: {:.4f}%".format(loss,accuracy * 100))
它显示print(type(Last_layer)) = <class 'keras.engine.keras_tensor.KerasTensor'>
我不知道为什么该行引用无类型对象
【问题讨论】:
标签: python tensorflow machine-learning keras deep-learning