当前位置: 首页 > 会展 > > 内容页

今日热议:ONNX模型部署利器ONNXRUNTIME框架

2023-06-17 23:23:48 来源: 面包芯语


(相关资料图)

GPU版本

然后配置库目录:

最后配置链接器,我的是支持CUDA版本,配置如下:

最后把DLL文件copy到编译生成的可执行文件同一个目录下,直接运行即可。C++推理,简单说分为四步,首先引入包含文件支持,再初始化推理会话对象Session,预处理图像,然后执行推理,最后完成后处理即可。

以ResNet18模型为例,导出ONNX格式,基于ONNXRUNTIME推理效果如下:

扫码查看OpenCV+OpenVIO+Pytorch系统化学习路线图

关键词:
x 广告
x 广告

Copyright ©  2015-2022 华中动漫网版权所有  备案号:京ICP备12018864号-26   联系邮箱:2 913 236 @qq.com