工具/软件:
您好:
我在其模型 zoo 中使用 TI 分布的以下网络:OD-8220_onnxrt_coco_edgeai-mmdet_yolox_s_lite_640x640_20220221_model_onnx。
我能够执行推理并成功地解析结果,但我不清楚在哪里可以找到有关解析输出张量的信息。
以随 zip 软件包分发的 onnx 文件为例 、我可以看到我们有一个1x3x640x640输入张量和两个输出张量:一个 float32的 dets 矩阵和一个 int64的标签数组。 然而,我发现我需要将所有输出张量解析为 float32以获得有意义的结果。
此外,我需要对 DETS 矩阵的列顺序进行逆向工程,以了解哪一个是分数,哪些是边界框的坐标。
问题是我是否缺少一些文档或了解如何从数据类型和数据内容的角度正确解析输入和输出张量的正确方法。 可以帮我解决这个问题吗?
谢谢