Ошибка при преобразовании фрагмента типа узла с помощью onnx_coreml

Я преобразовал свою модель даркнета YOLOv3-SPP в модель PyTorch .pt. Затем я преобразовал.pt в.onnx. Моя конечная цель - перейти к модели CoreML. Я пробовал использовать этот репозиторий GitHub. Однако при преобразовании моей модели я получаю такую ​​ошибку...

...    
145/229: Converting Node Type LeakyRelu
146/229: Converting Node Type Conv
147/229: Converting Node Type Reshape
148/229: Converting Node Type Transpose
149/229: Converting Node Type Reshape
150/229: Converting Node Type Slice
Traceback (most recent call last):
  File "convert2.py", line 11, in <module>
    coreml_model = convert(model_proto, image_input_names=['inputImage'], image_output_names=['outputImage'], minimum_ios_deployment_target='13')
  File "/usr/local/lib/python3.6/dist-packages/onnx_coreml/converter.py", line 626, in convert
    _convert_node_nd(builder, node, graph, err)
  File "/usr/local/lib/python3.6/dist-packages/onnx_coreml/_operators_nd.py", line 2387, in _convert_node_nd
    return converter_fn(builder, node, graph, err)
  File "/usr/local/lib/python3.6/dist-packages/onnx_coreml/_operators_nd.py", line 2011, in _convert_slice
    end_masks=end_masks
  File "/usr/local/lib/python3.6/dist-packages/coremltools/models/neural_network/builder.py", line 4220, in add_slice_static
    assert len(strides) == rank
AssertionError

Я использую такой сценарий...

import sys
from onnx import onnx_pb
from onnx_coreml import convert

model_in = sys.argv[1]
model_out = sys.argv[2]

model_file = open(model_in, 'rb')
model_proto = onnx_pb.ModelProto()
model_proto.ParseFromString(model_file.read())
coreml_model = convert(model_proto, image_input_names=['inputImage'], image_output_names=['outputImage'], minimum_ios_deployment_target='13')
coreml_model.save(model_out)

Этот простой скрипт python должен работать, но я не знаю, почему получаю эту ошибку. Я новичок в машинном обучении, поэтому не понимаю, как я могу даже попытаться решить эту проблему. Что мне делать, чтобы успешно преобразовать мою модель.onnx в CoreML?

1 ответ

Похоже, что ранг не совпадает между рангом входного тензора и параметром среза Не могли бы вы сообщить об ошибке на onnx-coreml

Как прокомментировал @matthijs-hollemans, попробуйте установить последнюю версию onnx-coreml

pip install onnx-coreml==1.2

Несколько других проблем:

  1. С какой версией модели onnx вы работаете? с Operator-9 поведение среза изменяется, и это может быть потенциальной точкой отказа преобразователя.

  2. Не могли бы вы также прикрепить модель ONNX?

Другие вопросы по тегам