终于进入正题了,首先我们先默认已经装好了onnx并且能正常运行,具体步骤跳到第一篇文章去看。
注意,我输入的是特征而不是图片,即输入是二维的不是三维的,三维的情况下篇文章再说。
下方是总体代码,我数据处理部分太长就不贴出来了,最后输入的特征是vector< float >的格式,具体一点就是8046 ,输出是20183,我的label类别有83,特征804卷出来后是201。python的onnx是可以任意长度的输入,不过c++的好像必须是固定长度,这些数字在之前python输出onnx模型的时候就设定好。
详细看代码内部的解析。
#include <assert.h>
#include<onnxruntime_cxx_api.h>
#include <iostream>
#include