专栏算法工具链模型多输入头,输入数据准备

模型多输入头,输入数据准备

已解决
luna2025-11-24
77
46

请问J6给onnx模型增加nv12转rgb的这个功能,支持5维多图像的输入吗,比如1x7x3x256x704,如果不支持除了在外部做预处理外,还有什么推荐的方式吗,比如拆分成7个1x3xx256x704的图像进行输入

在配置文件中发现这个功能好像只支持单输入的模型,请问如何处理呢

算法工具链
征程6
评论1
0/1000
  • Vincent
    Lv.4

    你好,你的原始onnx模型输入是咋样的?

    2025-11-24
    0
    45
    • luna回复Vincent:
      2025-11-24
      0
    • Vincent回复luna:

      现在是五维的,input_batch为1,要保证你所有的输入都是1开头的

      2025-11-24
      0
    • luna回复Vincent:

      其他的输入也都必须是1开头吗

      2025-11-24
      0
    • luna回复Vincent:

      我可以将imgs转为7个1x3xx256x704,4维的输入

      2025-11-24
      0
    • luna回复Vincent:

      疑惑的是还有其他输入头,input_layout_train应该如何写呢

      2025-11-24
      0
    • Vincent回复luna:

      也可以这样的 其他输入是多少,那个input_layout_train就怎么写 ,这个主要是准备校准集

      2025-11-24
      0
    • Vincent:

      拆成七个batch

      2025-11-24
      0
    • Vincent:

      input_batch那里不需要指定为7了 或者说 不需要这个参数了

      2025-11-24
      0
    • Vincent:

      会默认你的输入形状, 你只要按照输入形状准备你的校准集就行了 。

      2025-11-24
      0
    • luna回复Vincent:

      图像的input_layout_train是nchw,那其他输入是featuremap,这个地方我应该填写什么?

      2025-11-24
      0
    • Vincent回复luna:

      直接输入变成 7x3xHXW, NV12只支持四维 , 按照这个准备数据集 input_batch不需要显示制定了 , 其他的输入头形状该是多少是是多少

      2025-11-24
      0
    • Vincent回复luna:

      你的inpuyt_type_train是训练时候的数据类型,校准集也要按照训练时候的数据准备, input_layout_train,是onnx模型的数据排布 , 校准集都要保持统一的

      2025-11-24
      0
    • luna回复Vincent:

      那我维度不足的需要手动增加吗,需要保持每个输入头的第一个维度都是7吗

      2025-11-24
      0
    • Vincent回复luna:

      不需要的, 不过 你现在是想用featuremap格式还是bgr格式?

      2025-11-24
      0
    • Vincent回复luna:

      如果你都设置成featuremap,就是input_type_train,和input_type_rt的类型

      2025-11-24
      0
    • luna回复Vincent:

      imgs想用nv12输入,然后模型增加nv12转rgb的节点,其他的输入使用featuremap输入

      2025-11-24
      0
    • Vincent回复luna:

      img用nv12其他用featuremap,

      2025-11-24
      0
    • Vincent回复luna:

      input_typr_rt:NV12;featuremap;featuremap............. 这样指定

      2025-11-24
      0
    • Vincent回复luna:

      input_type_train也一样:bgr,featuremap;featuremap..............

      2025-11-24
      0
    • luna回复Vincent:

      好的,谢谢

      2025-11-24
      0
    • luna回复Vincent:
      2025-11-25
      0
    • luna回复Vincent:

      生成的是这样的

      2025-11-25
      0
    • Vincent回复luna:

      指定形状那里:7x3xhxw;featuremap形状;featuremap形状;................

      2025-11-25
      0
    • luna回复Vincent:

      那mean_value/scale_value的部分,featuremap需要指定什么吗

      2025-11-25
      0
    • Vincent回复luna:

      你这个1x7x3x256x784不是跟7x3x256x784一样嘛

      2025-11-25
      0
    • luna回复Vincent:

      其他输入的部分怎么指定, mean_value/scale_value的部分

      2025-11-25
      0
    • Vincent回复luna:

      设置成featuremap,就不需要预处理了,不需要指定了只指定预处理bgr的就可以了

      2025-11-25
      0
    • luna回复Vincent:

      好的

      2025-11-25
      0
    • Vincent回复luna:

      这个不需要预处理我意思是你提前处理好,到HBM模型转换节点nv12转bgr之后就不需要预处理了, 仅限于featuremap , 其他的bgr类型的你指定之后nv12转bgr模型内部会自动处理 , 纠正一下 ,featuremap对应的输入头不会转换节点,输入什么就是什么 。

      2025-11-25
      0
    • luna回复Vincent:

      那不需要预处理的输入头,归一化部分是空着就可以还是填什么值呢?

      2025-11-25
      0
    • Vincent回复luna:

      空着就可以

      2025-11-25
      0
    • luna回复Vincent:
      我把imgs配置了nv12和bgr,其余是配置featuremap,scale_value和mean_value只配置了img的三个通道值,具体如下
      2025-11-25
      0
    • luna回复Vincent:
      但是编译模型的时候会出现报错
      2025-11-25
      0
    • Vincent回复luna:

      mean_value,scale_value加分号试一下 ;;;;; 这样,featuremap对应的都填;;;

      2025-11-25
      0
    • luna回复Vincent:

      那img还是以3通道的格式写吧

      2025-11-25
      0
    • Vincent回复luna:

      是的

      2025-11-25
      0
    • luna回复Vincent:

      好像能够编译了,非常感谢

      2025-11-25
      0
    • luna回复Vincent:

      等下看看编译结果

      2025-11-25
      0
    • luna回复Vincent:
      插入转换节点的时候报错了,是有什么配置项没有加的原因吗
      2025-11-25
      0
    • Vincent回复luna:

      onnx是咋样的?要重新导出来的。onnx还是1x7x3xhxw嘛?

      2025-11-25
      0
    • luna回复Vincent:


      imgs输入头的shape已经改过来了,7x3x256x704
      2025-11-25
      0
    • luna回复Vincent:

      我给分解成7个1x3x256x704的输入头,然后各个的那个加预处理节点试试

      2025-11-25
      0
    • Vincent回复luna:

      NV12的输入似乎第一个维度只能为1

      2025-11-25
      0
    • luna回复Vincent:

      可以编译了,应该是分离模型输入头的问题,分离成七个单独的输入就没问题了 。

      2025-11-25
      0
    • luna:

      感谢解答

      2025-11-25
      0