Adding in the data first preprocessing, as well as the first model training.

2019-04-14 14:32:08 -05:00
parent 81a48a51ae
commit bc44d30180
11 changed files with 600 additions and 4 deletions
@@ -0,0 +1,12 @@
+from tensorflow 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)