Renaming files

This commit is contained in:
Lucas Oskorep
2019-04-14 19:02:36 -05:00
parent d3aac99da9
commit 97a30b0f48
11 changed files with 1 additions and 569 deletions
+12
View File
@@ -0,0 +1,12 @@
from tensorflow.contrib.keras.api import keras
from tensorflow.contrib import lite
keras_file = "weights.mobilenet.non-transfer.best.hdf5"
keras.models.load_model(keras_file)
h5_model = keras.models.load_model(keras_file)
converter = lite.TocoConverter.from_keras_model_file(keras_file)
tflite_model = converter.convert()
with open('mobilenet.tflite', 'wb') as f:
f.write(tflite_model)