跳转至主要内容
支持知识库

无法将转换后的自定义 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™ Processor Software OpenVINO™ toolkit Performance Libraries

免责声明

本页面上的内容是原始英文内容的人工翻译与计算机翻译的组合。我们提供此内容是为了您的便利并且仅供参考,未必完整或准确。如果本页面的英文版与翻译版之间存在任何冲突,应以英文版为准。 查看此页面的英语版本。