Add basic coreml and mlkit conversion scripts
This commit is contained in:
@@ -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__':
|
||||
|
||||
15
tensorflow_lite.py
Normal file
15
tensorflow_lite.py
Normal file
@@ -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')
|
||||
Reference in New Issue
Block a user