【AI工具链-模型转换】模型编译中input_source选项的配置方式

1.硬件获取渠道:公司采购

2.当前系统镜像版本:-

3.当前天工开物版本:hbdk version-3.31.1 horizon_nn version-0.13.9 hb_mapper version-1.8.7

4.问题定位:AI工具链hb_mapper makertbin模型转化

5.开发的demo/案例:-

6.需要提供的解决方案:

我目前通过hb_mapper makertbin进行模型转换,通过官方文档的描述,认为如果直接获取resizer输出的图像数据,需要在模型转换的input_source中配置resizer的选项,即:

因此我在配置文件config.yaml中增加了这个选项,模型输入节点的名字是image:0,此外我也尝试了其他的书写方式如"image:0": “resizer”,“image:resizer”,“resizer”等。


但是最终报错为:

因此,

①input_source在配置文件中的正确书写方式应该是怎样的。

②官方默认使用ddr中读取数据的方式是否代表并不推荐直接从pyramid或resizer中获取数据,是否还需要从pyramid或resizer将数据搬运至ddr中再输入模型?

input_source: {‘image:0’: ‘resizer’}