0

ダークネット 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 に正常に変換するにはどうすればよいですか?

4

1 に答える 1

0

入力テンソル ランクとスライス パラメータの間でランクが一致していないようですonnx-coremlにバグを報告してください

@matthijs-hollemans がコメントしたように、最新の onnx-coreml をインストールしてみてください

pip install onnx-coreml==1.2

その他の懸念事項:

  1. 使用している onnx モデルのバージョンは何ですか? Operator-9 ではスライスの動作が変更され、コンバーターの潜在的な障害点になる可能性があります。

  2. ONNXモデルも添付していただけますか？

于 2020-02-12T17:18:39.853 に答える