Lutz Roeder пре 7 година
родитељ
комит
6b379251ac
1 измењених фајлова са 20 додато и 0 уклоњено
  1. 20 0
      tools/onnx-converter.py

+ 20 - 0
tools/onnx-converter.py

@@ -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')