how to freeze the model

DetectionTeamUCAS / FPN_Tensorflow

This is a tensorflow re-implementation of Feature Pyramid Networks for Object Detection.

https://github.com/DetectionTeamUCAS/FPN_Tensorflow

MIT License

347 stars 132 forks source link

how to freeze the model #118

Closed shahmustafa closed 4 years ago

shahmustafa commented 4 years ago

Hi...how we can freeze the model to .pb?

I tried Tensorflow-object-detection's export_inference_graph.py script, it is throwing Key Conv/biases not found in checkpoint and to try TensorFlow's freeze_grapgh..py, it requires output node names

do you know any solution for this, @yangxue0827 @yangJirui

yangxue0827 commented 4 years ago

https://github.com/DetectionTeamUCAS/FPN_Tensorflow/tree/master/libs/export_pbs @shahmustafa