且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

用新课程重新训练模型

更新时间:2023-01-04 10:35:05

只需使用转移学习,然后创建一个新模型即可.

Just use a transfer learning, and create a new model.

model = VGG16(weights='imagenet',
                  include_top=False,
                  input_shape=(150, 150, 3))

model.pop()

base_model_layers = model.output
pred = Dense(11, activation='softmax')(base_model_layers)
model = Model(inputs=model.input, outputs=pred)

# Freeze the first layers, before train it
for layer in model.layers[:-2]:
    layer.trainable = False