Add basic coreml and mlkit conversion scripts

2021-03-29 18:58:16 +10:30
parent f3fc0f8fbb
commit f598005b73
2 changed files with 18 additions and 3 deletions
--- 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__':