文章 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 中有三個輸出節點,所有節點都需要在命令中指定:

模型優化指令: python mo.py --input_model model.onnx --輸出 --input_shape [1,3,512,512]

額外資訊

使用 Netron 視覺化 YOLOv5 ONNX 重量。

相關產品

本文章適用於 2 產品

這個頁面的內容綜合了英文原始內容的人工翻譯譯文與機器翻譯譯文。本內容是基於一般資訊目的,方便您參考而提供,不應視同完整或準確的內容。如果這個頁面的英文版與譯文之間發生任何牴觸,將受英文版規範及管轄。 查看這個頁面的英文版。