Add basic coreml and mlkit conversion scripts
This commit is contained in:
@@ -1,9 +1,9 @@
|
|||||||
import coremltools as ct
|
import coremltools as ct
|
||||||
|
|
||||||
|
|
||||||
def convert_coreml(saved_model_path):
|
def convert_coreml(model_path, save_path='../mobilenet_nnconv5.mlmodel'):
|
||||||
mlmodel = ct.convert(saved_model_path)
|
mlmodel = ct.convert(model_path)
|
||||||
mlmodel.save('../mobilenet_nnconv.mlmodel')
|
mlmodel.save(save_path)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
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