专栏算法工具链bin模型生成如何设定输入格式 在AI-EXPRESS中格式不一致导致HB_BPU_resize报错

bin模型生成如何设定输入格式 在AI-EXPRESS中格式不一致导致HB_BPU_resize报错

已解决
shaun2021-05-09
106
4

遇到这样一个问题,在AI-EXPRESS中,我参考yolov3_predict_method写了一个类似的预处理方法,但是在调用HB_BPU_resize的时候报错。

打印了结果发现主要原因是pre_resize_tensor的data_shape与input_tensor的data_shape维度不一致,pre_resize_tensor的data_shape为1xCxHxW,而我的input_tensor的data_shape为1xHxWxC,因此无法resize。

我想这里input_tensor与我的模型输入应该是一致的,但是模型编译过程中哪里可以控制input_layerout类型呢(NCHW/NHWC)?感觉应该是下面红框中的input_layerout_rt的类型。

请问一下红框内各个成分的含义是什么啊?然后由哪些参数决定的,我在config.yaml中好像没有发现如何控制这个参数。

另外请问如果我要调用usb_camera,我在生成bin模型时对input_type_rt有要求吗?(yolov3_predict_method中的类型是BPU_TYPE_IMG_NV12_SEPARATE)

感谢大佬解惑?

算法工具链
评论2
0/1000
  • 柏木木
    Lv.1

    我也遇到过这个问题,然后说分类示例模型里的 efficientnet-lite 系列都是 NHWC 输入的模型,我就根据那个里面的配置文件修改就好了,也不知道是哪个控制的输入

    2021-05-10
    0
    1
    • shaun回复柏木木:

      我的任务要求NCHW输入?

      2021-05-10
      0
  • admin
    Lv.1

    模型转化的时候input_type_rt 就是模型输入的类型,在config.ymal里面可以进行i需改,具体信息可以参照: 《常见问题FAQ手册》


    2021-05-10
    0
    1
    • shaun回复admin:

      发现了,输入格式与input_type_rt有关,上板要调用usb_camera的话用nv12类型比较妥

      2021-05-11
      0