Я пытаюсь преобразовать модель onnx в модель .ml для использования в ios. Моя конечная цель — использовать модель ml от Huggingface на iOS. На данный момент единственный способ преобразовать модель ml в файл модели .ml — использовать coremltools.converters. Я написал скрипт для преобразования модели Huggingface в onnx и onnx в mlmodel. Кажется, он конвертируется в onnx нормально, но затем возникает ошибка с coremltools.converters.onnx.convert. После некоторых исследований выяснилось, что вам нужно использовать только coremltools.convert, но я столкнулся с другой ошибкой. spec.ParseFromString(f.read()) google.protobuf.message.DecodeError: ошибка анализа сообщения с типом CoreML.Specification.Model
Вот как я конвертирую из onnx в coreml:
import coremltools
# print(coremltools.converters.onnx.convert)
# Load the ONNX model
onnx_model_path = "model.onnx"
onnx_model = coremltools.utils.load_spec(onnx_model_path)
model = coremltools.convert(onnx_model)
# Set the model's input and output descriptions (optional)
coreml_model.input_description["input"] = "Input image"
coreml_model.output_description["output"] = "Output class label"
# Save the Core ML model to disk
coreml_model.save("model.mlmodel")
Подробнее здесь: https://stackoverflow.com/questions/765 ... -to-coreml
