|
|
@@ -0,0 +1,20 @@
|
|
|
+#!/usr/bin/env python
|
|
|
+
|
|
|
+import os
|
|
|
+import sys
|
|
|
+
|
|
|
+file = sys.argv[1];
|
|
|
+base, extension = os.path.splitext(file)
|
|
|
+
|
|
|
+if extension == '.mlmodel':
|
|
|
+ import coremltools
|
|
|
+ import onnxmltools
|
|
|
+ coreml_model = coremltools.utils.load_spec(file)
|
|
|
+ onnx_model = onnxmltools.convert.convert_coreml(coreml_model)
|
|
|
+ onnxmltools.utils.save_model(onnx_model, base + '.onnx')
|
|
|
+elif extension == '.h5':
|
|
|
+ import keras
|
|
|
+ import onnxmltools
|
|
|
+ keras_model = keras.models.load_model(file)
|
|
|
+ onnx_model = onnxmltools.convert.convert_keras(keras_model)
|
|
|
+ onnxmltools.utils.save_model(onnx_model, base + '.onnx')
|