如何在张量流中删除最后一层预训练模型

Question

I want to remove the last layer of sports1m_finetuning_ucf101.model in C3D tensorflow ( https://github.com/hx173149/C3D-tensorflow ) and add my own classifier layer for my dataset which has different number of labels than the model. I know in Keras i can use the pop method but here I don't know how to load the model in tensorflow and remove the last layer. Is there anyone to tell me how I can do it?

Answer 1

In tensorflow editing model is not recommended. What you can probably do is that you can set the output based on the layer name.

for layer in model.layers:
    if layer.name == desired_layer_name:
        output = layer.output
        break