专栏算法工具链onnx模型浮点转定点时图片不匹配输入通道的问题

onnx模型浮点转定点时图片不匹配输入通道的问题

已解决
柏木木2021-04-12
85
1

我在试图将自己的onnx模型转量化模型时遇到如下报错, 看错误提示猜测应该是因为我的输入节点是NHWC, 而ERROR上一行中original_input_layout=NCHW, 所以说不匹配, 但我找不到哪里能改这个参数.

校准图片集的大小是准确的, 并且我在preprocess_on:True和做过预处理且preprocess_on:False的两种情况都试过,都会报这个错误.

请问如何解决?

算法工具链
评论1
0/1000
  • 芯链情报局
    Lv.4

    你好，可以参考下我们分类示例模型里的 efficientnet-lite 系列，都是 NHWC 输入的模型

    2021-04-12
    0
    0