Adding in new trained models, as well as new tools for reviewing the results.

Added in testing flow for testing our unfininshed/finished models.

Also adding a test dataset with one picture of every pokemon in the game.
This commit is contained in:
Lucas Oskorep
2019-04-30 13:00:24 -05:00
parent 601f4f7633
commit fbac5d519a
17 changed files with 1643 additions and 73 deletions
+12
View File
@@ -0,0 +1,12 @@
from tensorflow.contrib.keras.api import keras
from tensorflow.contrib import lite
keras_file = "mobilenetv2.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('mobilenetv2.tflite', 'wb') as f:
f.write(tflite_model)