diff --git a/coreml.py b/coreml.py index 6c03dc6..db2fbae 100644 --- a/coreml.py +++ b/coreml.py @@ -1,9 +1,9 @@ import coremltools as ct -def convert_coreml(saved_model_path): - mlmodel = ct.convert(saved_model_path) - mlmodel.save('../mobilenet_nnconv.mlmodel') +def convert_coreml(model_path, save_path='../mobilenet_nnconv5.mlmodel'): + mlmodel = ct.convert(model_path) + mlmodel.save(save_path) if __name__ == '__main__': diff --git a/tensorflow_lite.py b/tensorflow_lite.py new file mode 100644 index 0000000..c70c454 --- /dev/null +++ b/tensorflow_lite.py @@ -0,0 +1,15 @@ +import tensorflow as tf + + +def convert_tensorflow_lite(model_path, save_path='../mobilenet_nnconv5.tflite'): + # Convert the model + converter = tf.lite.TFLiteConverter.from_saved_model(model_path) # path to the SavedModel directory + tflite_model = converter.convert() + + # Save the model. + with open('model.tflite', 'wb') as f: + f.write(tflite_model) + + +if __name__ == '__main__': + convert_tensorflow_lite('../mobilenet-nnconv5-e12-experimental')