模型转换问题

应用开发 算法工具链
, , ,
1

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

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吗？

2

感谢您使用地平线芯片算法工具链，最近我们在收集大家的满意度反馈，欢迎您填写问卷，详细情况可见：https://developer.horizon.ai/forumDetail/146177053698464782

3

您好。hbDNNDGetInputCount用来获取模型有几个输入，而不是batch_size。batch模型推理的示例代码，可以查看：horizon_runtime_sample/code/02/nv12_batch，相关教学可以查看：多batch模型使用与部署 (horizon.ai)

4

好的，谢谢