pytorch 33 使用mmdeploy将mmdetection模型转换为onnx,并基于mmdeploy-sdk实现c++部署

可以将

格式,让

在不同的平台和框架之间进行

和迁移。同时,

c语言进行

,可以将

格式的

到C/

环境中。 要

,首先需要安装

cv-full和

。安装完成后,

以下命令将

python tools/deploy/export_</p> <p>.py ${CONFIG_FILE} ${CHECKPOINT_FILE} ${</p> <p>_FILE} --input-img ${IMG_SHAPE} 其中${CONFIG_FILE}为

的配置文件,${CHECKPOINT_FILE}为训练好的

文件,${

_FILE}为生成的

文件名称,${IMG_SHAPE}为输入图像的形状。

后,可以

。首先需要在C环境下

的API读取

文件,然后

C语言的库函数对

进行推理。

的API读取

文件,代码如下: #include</p> <p>.h" int main(int argc, char *<em>argv) { flexbuffer </em>model;</p> <p>session <em>session; const char </em>model_path = argv[1]; /* 读取</p> <p>文件 <em>/ model = read_flexbuffer_file(model_path); assert(model); /</em> 创建session <em>/ session = create_session(model); assert(session); /</em> 设置输入数据 <em>/ set_input_data(session, input_data); /</em> 进行推理,获取输出数据 */ get_output_data(session, output_data); return 0; }

C语言的库函数对

进行推理,主要的工作是对输入数据进行预处理,并调用session的run方法进行推理,代码如下: #include</p> <p>.h" int main(int argc, char *<em>argv) { flexbuffer </em>model;</p> <p>session <em>session; float </em>input_buf, <em>output_buf; /</em> 预处理输入数据 <em>/ input_buf = preprocess_input(input_data); /</em> 读取</p> <p>文件 <em>/ model = read_flexbuffer_file(argv[1]); assert(model); /</em> 创建session <em>/ session = create_session(model); assert(session); /</em> 设置输入数据 <em>/ set_input_data(session, input_data); /</em> 进行推理,获取输出数据 <em>/ get_output_data(session, output_data); /</em> 对输出数据进行后处理 */ output_buf = postprocess_output(output_data); return 0; }

进行C语言

时,需要注意

的输入和输出数据类型和形状,以及预处理和后处理函数的编写。通过以上步骤,就可以

C语言

Original: https://blog.csdn.net/a486259/article/details/126657477
Author: 万里鹏程转瞬至
Title: pytorch 33 使用mmdeploy将mmdetection模型转换为onnx,并基于mmdeploy-sdk实现c++部署

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/708499/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

  • .Net6 之 asp.net core webapi Swagger 版本控制及接口注释说明

    只有 一个 版本的webapi 配置如下 第一步 首先 在 控制器上 和 每个接口方法 上 添加 注释 如图 第二步 :打开项目的.csproj文件加上 添加 如图: 第三步: 在…

    人工智能 2023年6月28日
    0101
  • 2021中国大学MOOC 机器学习(温州大学) 最新中国大学MOOC满分章节测试答案

    引言 引言课后测试1、 问题:哪一个是机器学习的合理定义?选项:A:机器学习从标记的数据中学习B:机器学习能使计算机能够在没有明确编程的情况下学习C:机器学习是计算机编程的科学D:…

    人工智能 2023年6月30日
    083
  • 【语音识别】基于matlab矢量量化(VQ)说话人识别【含Matlab源码 575期】

    ⛄一、获取代码方式 获取代码方式1:完整代码已上传我的资源:【语音识别】基于matlab矢量量化(VQ)说话人识别【含Matlab源码 575期】点击上面的蓝色字体,付费直接下载,…

    人工智能 2023年5月27日
    086
  • (WSI分类)WSI分类文献小综述

    今天介绍一些WSI分类的文献。涉及三种比较主要的技术:传统的特征分类,CNN和Transformer。 传统特征WSI分类 Classification of Tumor Hist…

    人工智能 2023年7月1日
    064
  • 商业数据的加工处理

    目录 商业数据的加工 抽取数据源中的指定数据 数据源的分类汇总(分组) 完善数据源 透视列与逆透视列 商业数据的整理与转换 查询编辑器 数据排序和筛选 数据源合并与转换 数据合并的…

    人工智能 2023年7月16日
    063
  • Mask2former

    啊哦~你想找的内容离你而去了哦 内容不存在,可能为如下原因导致: ① 内容还在审核中 ② 内容以前存在,但是由于不符合新 的规定而被删除 ③ 内容地址错误 ④ 作者删除了内容。 可…

    人工智能 2023年5月26日
    074
  • 基于情绪因子的CTA截面策略

    量化投资与机器学习微信公众号,是业内垂直于 量化投资、对冲基金、Fintech、人工智能、大数据等领域的主流自媒体。公众号拥有来自 公募、私募、券商、期货、银行、保险、高校等行业 …

    人工智能 2023年6月11日
    097
  • 一文读懂机器学习分类全流程

    目录 前言 提出问题 一、介绍 1.分类简介 2.imblearn的安装 二、数据加载及预处理 1.加载并查看数据 ①导入Python第三方库 ②调用并查看数据 2.查看数据分布 …

    人工智能 2023年7月4日
    067
  • Pytorch_lstm详细讲解

    1.详细讲解官方文档的例子: 这里有个老哥先带你回顾一下lstm的理论知识: pytorch中lstm参数与案例理解。_wangwangstone的博客-CSDN博客_torch….

    人工智能 2023年7月20日
    085
  • 补码 符号数的二进制 整数

    Blablabla~ 今天重温了《深入了解计算机系统》关于符号数的bianary表示,弄懂了补码的原理。并写了python脚本验证。当初对这块有疑惑是一上来就告诉你,xxx取反加1…

    人工智能 2023年6月28日
    0109
  • cs224w(图机器学习)2021冬季课程学习笔记8 Colab 2

    诸神缄默不语-个人CSDN博文目录cs224w(图机器学习)2021冬季课程学习笔记集合 文章目录 1. PyG包的Dataset和Data 2. ogb包介绍 3. 节点预测任务…

    人工智能 2023年6月15日
    0100
  • R实战 | 倾向性评分匹配(PSM)

    PSM 「倾向性评分匹配」(propensity score matching,PSM)是一种用来评估处置效应的统计方法。广义说来,它将样本根据其特性分类,而不同类样本间的差异就可…

    人工智能 2023年6月15日
    0123
  • 集显安装pytorch教程

    集显看最下面 首先GPU安装教程 1.安装Anaconda(这个是可以安装环境和前置的软件),这个网址是清华源下载,可以去官网下载 Index of /anaconda/archi…

    人工智能 2023年7月22日
    066
  • 在部署过程中如何处理模型的更新和版本控制

    问题背景: 在机器学习模型的部署过程中,我们经常需要更新模型以提升性能或修复bug。同时,为了能够追踪模型的版本历史,并保持模型和代码的一致性,我们也需要进行模型的版本控制。本文将…

    人工智能 2024年1月3日
    051
  • AutoDL-GPU租用平台使用教程

    AutoDL是一个国内的GPU租用平台,最近使用了一下,体验感还是很nice的,所以写了篇博客来介绍一下该平台的具体使用方式,也可以当做一个教程来学习^_^ 网址:AutoDL-品…

    人工智能 2023年7月4日
    099
  • Python数据分析报告

    业务背景:B2B业务已收会员和服务费作为收益主要来源,目前会员类型分成钻石会员、金牌会员,销售合同一般会包含会员服务、增值服务、广告服务等等,销售过程可能会受到销售策略的影响,做一…

    人工智能 2023年7月18日
    066
亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球