coreml-converter.py 256 B

123456789101112
  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 == '.h5':
  7. import coremltools
  8. coreml_model = coremltools.converters.keras.convert(file)
  9. coreml_model.save(base + '.mlmodel')