专栏算法工具链执行hb_mapper makertbin的过程出现reshape节点执行失败报错

执行hb_mapper makertbin的过程出现reshape节点执行失败报错

已解决
Linxiaoxu2022-08-13
40
1

docker : hub.hobot.cc/aitools/ai_toolchain_centos_7_xj3: v1.13.3

在hb_mapper makertbin过程出现了reshape失败的报错

浮点模型的输入为[2,3,1024,1024],在yaml文件中也设置了input_batch为2.

这里的报错与原因似乎是在calibration的过程中,程序给模型只传入了一张图片,导致reshape节点执行失败

使用netron对模型进行可视化,发生错误的节点为如下图中的reshape节点

请问该如何解决

算法工具链
评论1
0/1000
  • Horizon_guosheng
    Lv.1

    您好,batchsize目前仅支持1,不支持其他数值

    2022-08-13
    0
    0