tensorflow - 在 Ort:Value:CreateTensor 中将张量排序为线性数组

Question

这看起来很基本，但由于某种原因，我找不到任何明确的文档。

所以可以说我知道我的 ONNX 模型需要一个 shape 输入[245, 245, 3]。构造函数中的第二个参数Ort::Value::CreateTensor需要一个线性数据数组来填充张量。线性阵列的顺序是什么？

例如，线性数组中的前三个值是图像中第 0 个像素的 BGR 值，还是线性数组中的前三个值是图像中前三个像素的 B 通道值？至于图像中像素的排序：行主要？

Answer 1

这是行主要的。ONNX 中的输入格式是 NCHW。C = 通道数。在这种情况下，C=3。C（BGR 或 RGB）的顺序取决于模型。例如，YOLO 模型采用 3(RGB) x 416px x 416px 的图像。