0
点赞
收藏
分享

微信扫一扫

c++动态库之四 onnxruntime

善解人意的娇娇 2022-03-22 阅读 79
c++

终于进入正题了,首先我们先默认已经装好了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 
举报

相关推荐

0 条评论