专栏算法工具链yolov5量化提示找不到MaxPool

yolov5量化提示找不到MaxPool

已解决
zc2022-02-28
116
1

采用hb_mapper export量化自己训练的yolov5.onnx模型的时候,出现下面问题:

官网算子支持与约束列表中明确包含支持MaxPool这个算子。

而且很奇怪的是,model_zoo示例中提供的yolov5s.onnx中也包含有MaxPool,但我量化这个模型时,却没有出现上述错误。

请问这是什么原因?

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    括号中表示的是 opset 版本,目前地平线支持的是 opset10/11 的核心算子,opset12 超出范围了,建议导出 onnx 时指定为 opset10/11

    2022-03-01
    2
    0