使用训练有素的体重训练不同的数据集

Question

我已经在包含4个输出类的数据集中训练了CNN模型，我将模型的权重保存在“ weights.h5”中。我想使用这些权重在仅包含2个输出类的另一个数据集上进行训练，该怎么做？

Answer 1

1. 加载模型/重量
2. 从受过训练的网络中删除最后一层
3. 附加新的输出层
4. 重新编译

base_model.load_weights(...)

x = base_model.layers[-2].output
x = Dense(2,activation='softmax')(x) # 2 output classes
model = Model(base_model.input,x)

model.compile(...)