Skip to content
Snippets Groups Projects
Commit e189ae60 authored by ch1u20's avatar ch1u20
Browse files

-Tidying up tflite conversion code

parent 6ca44b6c
No related branches found
No related tags found
No related merge requests found
......@@ -54,24 +54,33 @@ def Evaluate_TUT2017_With_Model(model_file_path):
#ModelFunctions.Generate_Model_Graphs(X_test, model_file_path, __models_dir)
#ModelFunctions.Generate_Ground_Truth_Graphs(Y_test, __models_dir)
if __name__ == '__main__':
#for i in range(100):
#Evaluate_TUT2017_With_Model("/Users/charles/Documents/MINDS/Year1/6003_Project/local_testing/models_from_iridis/Model_CH_12/Model_CH_12__05_06_23_58_09.tf")
#Evaluate_TUT2017_With_Model("/Users/charles/Documents/MINDS/Year1/6003_Project/local_testing/models_from_iridis/Model_ES_05/Model_ES_05__04_22_01_34_08.tf")
#Evaluate_TUT2017_With_Model("/mainfs/cdt/models/Model_CH_12/Model_CH_12__05_06_23_58_09.tf")
def convert_to_tflite():
binary_model = "/home/charles/Documents/MINDS/Year1/Interdisciplinary_Project_6003/PyProjects/localtesting/models/Model_CH_12/Model_CH_12__05_06_23_43_52.tf"
fp_model = "/home/charles/Documents/MINDS/Year1/Interdisciplinary_Project_6003/PyProjects/localtesting/models/Model_CH_13/Model_CH_13__05_19_17_45_00.tf"
#Load the Model
saved_model = tf.keras.models.load_model(binary_model)
saved_model_binary = tf.keras.models.load_model(binary_model)
saved_model_fp = tf.keras.models.load_model(fp_model)
# Converting a tf.Keras model to a TensorFlow Lite model.
#converter = tf.lite.TFLiteConverter.from_keras_model(saved_model)
#tflite_model = converter.convert()
converter_fp = tf.lite.TFLiteConverter.from_keras_model(saved_model_fp)
tflite_model_fp = converter_fp.convert()
import larq_compute_engine
tflite_model = larq_compute_engine.convert_keras_model(saved_model)
tflite_model_binary = larq_compute_engine.convert_keras_model(saved_model_binary)
# Save the model.
# Save the binary model.
with open('model_binary.tflite', 'wb') as f:
f.write(tflite_model)
\ No newline at end of file
f.write(tflite_model_binary)
# Save the fp model.
with open('model_fp.tflite', 'wb') as f:
f.write(tflite_model_fp)
if __name__ == '__main__':
#for i in range(100):
#Evaluate_TUT2017_With_Model("/Users/charles/Documents/MINDS/Year1/6003_Project/local_testing/models_from_iridis/Model_CH_12/Model_CH_12__05_06_23_58_09.tf")
#Evaluate_TUT2017_With_Model("/Users/charles/Documents/MINDS/Year1/6003_Project/local_testing/models_from_iridis/Model_ES_05/Model_ES_05__04_22_01_34_08.tf")
#Evaluate_TUT2017_With_Model("/mainfs/cdt/models/Model_CH_12/Model_CH_12__05_06_23_58_09.tf")
convert_to_tflite()
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment