diff --git a/5 - KerasToTensorflow.py b/5 - KerasToTensorflow.py index 8ff6943..bd64c91 100644 --- a/5 - KerasToTensorflow.py +++ b/5 - KerasToTensorflow.py @@ -1,7 +1,7 @@ from tensorflow.contrib.keras.api import keras from tensorflow.contrib import lite -keras_file = "InceptionV3Full.hdf5" +keras_file = "mobilenet.hdf5" keras.models.load_model(keras_file) h5_model = keras.models.load_model(keras_file)