| 1234567891011121314151617181920 |
- #!/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')
|