擅长:python、mysql、java
<p>您已经使用了预先训练的ResNet的完全连接层,您需要创建适合您的任务的适当分类层</p>
<pre class="lang-py prettyprint-override"><code>from tensorflow.keras.layers import GlobalAveragePooling2D
from tensorflow.keras import Model
model = ResNet50(include_top=False)
f_flat = GlobalAveragePooling2D()(model.output)
fc = Dense(units=2048,activation="relu")(f_flat)
logit = Dense(units=5, activation="softmax")(fc)
model = Model(model.inputs,logit)
</code></pre>