Updating the input image sizes to have a higher resolution. Currently unclean data is reaching 70% accuracy.

2019-04-14 18:50:53 -05:00
parent aabd88d4fb
commit de117a55ae
23 changed files with 1869 additions and 7 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)