Ошибка при преобразовании фрагмента типа узла с помощью 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
Несколько других проблем:
С какой версией модели onnx вы работаете? с Operator-9 поведение среза изменяется, и это может быть потенциальной точкой отказа преобразователя.
Не могли бы вы также прикрепить модель ONNX?