scrpit to freeze keras model file and inference example file

qubvel / efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

https://arxiv.org/abs/1905.11946

Apache License 2.0

2.07k stars 472 forks source link

Closed ghost closed 1 year ago

ghost commented 4 years ago

Added a script to freeze Keras model file (h5) into the Tensorflow model file (pb). Also added example ipython file for frozen model inference.

UmarSpa commented 4 years ago

Is it possile to freeze only the first 2 layers ?

ghost commented 4 years ago

Yes, I guess it is possible. Please have a look at this link.