专栏算法工具链.horizon_onnxruntime.onnxruntime 出错

.horizon_onnxruntime.onnxruntime 出错

已解决
海平面ovo2023-02-24
45
3

用户您好，请详细描述您所遇到的问题，这会帮助我们快速定位问题~

1.芯片型号：例如X3、X3派、J5等
2.天工开物开发包OpenExplorer版本：例如XJ3_OE_2.4.2，XJ3_OE_1.15.2，J5_OE_1.1.40等
3.问题定位：例如模型转换、板端部署等
4.问题具体描述：请提供运行的命令、报错信息，如果可以的话，可在附件中提供模型供技术支持进行复现

horizon_nn.horizon_onnxruntime.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Unexpected input data type. Actual: (N11onnxruntime17PrimitiveDataTypeIhEE) , expected: (N11onnxruntime17PrimitiveDataTypeIfEE)

这部分 给模型 输入数据有格式要求吗？ 普通的 numpy (NCHW） 可以吗? 这部分有例子吗？

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

    您好，请问您使用的的芯片类型和OE版本是什么呢？是发生在什么阶段的报错呢？直接给这样的报错提示，是没办法定位问题的哈，不是很理解您的需求

    2023-02-24
    0
    1
    • 海平面ovo回复颜值即正义:

      问题解决了， img_tensor = img.transpose(2,0,1)[NULL].astype('float32')， 输入后面应该转成 float32,

      (N11onnxruntime17PrimitiveDataTypeIhEE) , expected: (N11onnxruntime17PrimitiveDataTypeIfEE，这儿提示了

      2023-02-27
      0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0