跳到主要內容
支援知識庫

無法將自訂 YOLOv5 ONNX 型號優化為 INT8 中間表示 （IR）

內容類型: 錯誤訊息   |   文章 ID: 000092780   |   最近查看日期: 2022 年 11 月 16 日
描述 解決方法

描述

  • 將自訂 YOLOv5 ONNX 模型轉換為 FP32 IR：
    mo --input_model model.onnx
  • 使用 FP32 IR 執行 Post 訓練優化工具 （POT）。
  • 收到錯誤：
    Error: AssertionError: Incorrect mapping then_graph outputs with Result_6617 outputs! Can't find port with ID 4 in If operation.

解決方法

  1. 在Netron 中視覺化ONNX 模型。
  2. 搜尋 Netron 中的關鍵字 Transpose。
  3. 取得 回流節點名稱。
  4. 在 將 ONNX 模型轉換為 IR 時，指定回流節點名稱：
    mo --input_model model.onnx --output
  5. 運行 含 IR 的鍋：
    pot -c model.json

相關產品

本文章適用 3 產品。
Intel® Xeon Phi™ 處理器軟體 OpenVINO™ toolkit 效能程式庫

免責聲明

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

需要更多協助嗎？

聯絡支援部門
聯絡支援部門