专栏算法工具链模型转换问题

模型转换问题

已解决
fgz9992023-03-06
67
3

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

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

问题:原始模型输入shape是1x3xHxW,若转换时设置input_batch=2,转换后的.bin模型的输入shape是2x3xHxW吗?在runtime示例中,分别加载转换时设置的input_batch=1和input_batch=2生成的.bin模型,发现通过hbDNNDGetInputCount()函数得到的input_count值都是1,按理说模型输入第一维不应该分别是1和2吗?
算法工具链
征程3征程5
评论2
0/1000
  • 颜值即正义
    Lv.2
    您好。hbDNNDGetInputCount用来获取模型有几个输入,而不是batch_size。batch模型推理的示例代码,可以查看:horizon_runtime_sample/code/02/nv12_batch,相关教学可以查看:多batch模型使用与部署 (horizon.ai)
    2023-03-06
    0
    1
    • fgz999回复颜值即正义:

      好的,谢谢

      2023-03-06
      0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0