問題タブ [onnxruntime]

最後に、最初の質問をします。私は過去数日間これに苦労してきましたが、ここで同じ質問を見つけることができませんでした.

次のコードを使用して、この事前トレーニング済みモデルを ONNXに変換しました。

midas.onnx を Netron で開いたところ、問題なく表示されました。そして、onnxruntime でテストしようとしました:

しかし、私はこのエラーが発生しました:

詳細については、Pytorch (1.3.1)、ONNX (1.7.0)、および Onnxruntime (1.3.0) を使用します。

助けていただければ幸いです。

ONNX 形式のモデルが 2 つあります。どちらのモデルも似ています (どちらも事前トレーニング済みの深層学習モデルです。例: ResNet50 モデル)。それらの唯一の違いは、最後のレイヤーが異なるデータセット用に最適化/再トレーニングされることです。

以下に示すように、これら 2 つのモデルの最初の k レイヤーをマージします。これにより、推論のパフォーマンスが向上するはずです。

私のケースを明確にするために、これらはこの機能を実装する他の機械学習ツールの例です。パイトーチ、ケラス。

sklearn SVC モデルを .onnx ファイルとして変換/保存しようとしていますが、理解できないランタイム エラーが発生します。sklearn ランダム フォレスト分類器と sklearn k-NN 分類器を使用して、この同じコードをエラーなしで効果的に使用することができました。理解できない奇妙な onnx ランタイム エラーが発生します。このエラーに関するヘルプをいただければ幸いです。

以下に、最初にファイル svm_time.py を実行した出力を投稿し、その下に svm_time.py ファイルに含まれるコードを含めました。

ありがとう。