擅长:python、mysql、java
<p>我找到了这个解决方案,它对我有效</p>
<pre><code>def Create_Model():
#input_shape = (150,150,3)
model_1 = VGG16(weights='imagenet',include_top=False)
input = keras.layers.Input(shape=(150,150,3))
Last_layer=model_1(input)
Model_Vgg=keras.layers.Flatten()(Last_layer)
#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 =input,outputs = Model_Vgg )
return model
</code></pre>