onnx-converter.py 569 B

1234567891011121314151617181920
  1. #!/usr/bin/env python
  2. import os
  3. import sys
  4. file = sys.argv[1];
  5. base, extension = os.path.splitext(file)
  6. if extension == '.mlmodel':
  7. import coremltools
  8. import onnxmltools
  9. coreml_model = coremltools.utils.load_spec(file)
  10. onnx_model = onnxmltools.convert.convert_coreml(coreml_model)
  11. onnxmltools.utils.save_model(onnx_model, base + '.onnx')
  12. elif extension == '.h5':
  13. import keras
  14. import onnxmltools
  15. keras_model = keras.models.load_model(file)
  16. onnx_model = onnxmltools.convert.convert_keras(keras_model)
  17. onnxmltools.utils.save_model(onnx_model, base + '.onnx')