专栏算法工具链LSTM训练数据应该如何进行校准

LSTM训练数据应该如何进行校准

已解决
红鲤鱼绿鲤鱼与驴2026-05-19
70
5

各位大佬好:

1. 我使用OE1.16.6的xj3工具包,现在已经将LSTM网络转为onnx文件,在第二步校准excel中的数据的时候应该如何进行处理来得到校准数据,麻烦解答一下吧,谢谢啦!

2. 如果我的网络本身输入就是两张图象的时候 例如双目网络的输入为两张图象,处理方法是不是与多batch推理相同,在使用多batch推理的时候能不能显著提高显存利用率呢,感谢各位!

算法工具链
技术深度解析征程3
评论3
0/600
  • YCJ
    Lv.4
    LSTM 的校准数据需要按照 ONNX 模型真实输入 tensor 来准备。Excel 只是原始数据源,不能直接作为校准输入。建议先确认 ONNX 输入 name、shape 和 dtype,然后将 Excel 中的特征列按训练/推理时一致的前处理方式处理,并按 LSTM 的输入维度组织好,最后保存为 float32 的 .bin 或 .npy 作为校准数据。
    2026-05-19
    0
    2
    • 红鲤鱼绿鲤鱼与驴回复YCJ:

      感谢佬儿的解答,还想问问有没有相关的数据处理实例能够提供一下参考,谢谢

      2026-05-20
      0
    • YCJ回复红鲤鱼绿鲤鱼与驴:

      校准集准备的原理都是一样的,OE包里有很多其他模型准备校准集的示例可以参考

      2026-05-20
      0
  • YCJ
    Lv.4
    双目网络输入两张图像和多 batch 推理不是同一个概念。双目网络的两张图通常是同一个样本的左目/右目输入,模型内部会利用两张图之间的对应关系进行特征匹配或融合;而多 batch 是一次推理多个相互独立的样本。
    因此,双目模型的数据准备方式要以 ONNX 的输入定义为准。如果 ONNX 有两个输入,就需要分别准备 left/right 两路输入,并保证样本一一对应
    2026-05-19
    0
    0
  • HuangHui
    Lv.5

    客户您好,长时间未收到你的答复,相信问题已解。如对此尚存疑问欢迎新帖讨论,感谢您的参与!

    2026-07-03
    0
    0