記事 ID: 000088394 コンテンツタイプ: エラーメッセージ 最終改訂日: 2022/05/20

YOLOv5 ONNX モデルを中間表現 (IR) 形式に変換する際に受信したノード関連のエラー

BUILT IN - ARTICLE INTRO SECOND COMPONENT
概要

YOLOv5 ONNX モデルを IR 形式に変換するモデル・オプティマイザーのコマンドを修正します。

詳細
  • YOLOv5 ONNX モデルを IR 形式に変換:

    python /opt/intel/openvino_2021/deployment_tools/model_optimizer/mo.py --input_model "model.onnx" --output_dir  "cur_folder/"

  • 受信エラー:

    [ ERROR ] The ExpandDims node Unsqueeze_583 has more than 1 input
    [ ERROR ] Cannot infer shapes or values for node "Slice_4".
    [ ERROR ] Output shape: [0 3 0 0] of node "Slice_4" contains non-positive values
    [ ERROR ] It can happen due to bug in custom shape infer function .
    [ ERROR ] Or because the node inputs have incorrect values/shapes.
    [ ERROR ] Or because input shapes are incorrect (embedded to the model or passed via --input_shape).

解決方法

YOLOv5 には 3 つの出力ノードがあり、それらのノードすべてをコマンドで指定する必要があります。

Model Optimizer コマンド: Python mo.py --input_model model.onnx --output --input_shape [1,3,512,512]

追加情報

Netron使用して、YOLOv5 ONNX ウェイトを視覚化します。

このページのコンテンツは、元の英語のコンテンツを人力翻訳および機械翻訳したものが混在しています。この内容は参考情報および一般的な情報を提供するためものであり、情報の正確さと完全性を保証するものではありません。インテルは不正確な翻訳があった場合でもいかなる責任を負いません。このページの英語版と翻訳の間に矛盾がある場合は、英語版に準拠します。 このページの英語版をご覧ください。