ATC模型转换

2023-05-16

(1)在使用昇腾硬件(例如Atlas200)进行模型推理的时候,首先需要使用ATC工具将原始模型转换为适合昇腾硬件的模型(.om),支持原始框架类型为Caffe、TensorFlow、MindSpore、ONNX的模型转换,其中ONNX可以看做是一个中间模型,如果其他模型可以转换为ONNX模型那么也可以在使用ATC进行模型转换,例如pytorch模型转换为ONNX模型,ONNX模型再转换为.om文件。

(2)常用参数解释:

--model:原始模型文件路径与文件名

--weight:权重文件路径与文件名

--framework:原始模型框架:0是caffe,1是mindspore,3是tensorflow,5是onnx

--input_format:输入数据格式:Caffe默认为NCHW,TensorFlow默认为NHWC

--dynamic_image_size:设置输入图片的动态分辨率参数。适用于执行推理时,每次处理图片宽和高不固定的场景。

--output:存放转换后的离线模型的路径以及文件名

--soc_version:模型转换时指定芯片版本。

--insert_op_conf:插入算子的配置文件路径与文件名,例如aipp预处理算子(包含了图像预处理比如色域转换、减均值等)。

(3)例子:atc --model=yolov3.prototxt --weight=yolov3.caffemodel --framework=0 --output=yolov3_framework_caffe_aipp_1_batch_1_input_int8_output_FP32 --soc_version=Ascend310 --insert_op_conf=aipp_nv12.cfg

对模型yolov3.prototxt和权重yolov3.caffemodel的caffe原始模型进行转换,因此framework为0表示caffe,输出文件名字为yolov3_framework_caffe_aipp_1_batch_1_input_int8_output_FP32,推理芯片选择Ascend310,算子配置文件为aipp_nv12.cfg。

PS:CANN训练营第三期正在进行中,这次训练营包含了模型营、应用营、算子营。基本包涵了华为昇腾AI全流程开发的各个环节,欢迎感兴趣的同学报名参加~报名地址:昇腾CANN训练营第三期_开发者-华为云

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

ATC模型转换 的相关文章

随机推荐

  • vue安装vue-router出错

    项目场景 xff1a 在vue中安装vue router 问题描述 xff1a 提示 xff1a 在安装过程中报错 xff0c 缺少依赖 xff1a PS D span class token punctuation span WebDep
  • SpringBoot之SpringSecurity(安全)

    SpringSecurity xff08 安全 xff09 Spring Security是针对Spring项目的安全框架 xff0c 也是Spring Boot底层安全模块默认的技术选型 xff0c 他可以实现强大的Web安全控制 xff
  • 解决决SpringBoot图片上传需重启服务器才能显示的问题

    问题描述 xff1a 前端页面实现头像图片上传并实时更新显示的功能 xff0c 但是文件上传成功后不能实时显示 xff0c 必须重启服务器后才能显示出来 下图是我将图片文件上到到的目录 xff0c 及项目架构 问题发生的原因 这是服务器的自
  • html页面实现登录验证码功能(纯前端)

    html 43 css 43 javascript简易实现注册或登录时实现验证码功能 xff1a 原文链接 xff1a https blog csdn net weixin 41472431 article details 90732468
  • lvs2种模式和haproxy配置

    lvs配置方法 nat模式配置 ip类型ip地址VIP仅主机模式192 168 96 134DIP192 168 149 143host1IP192 168 149 140host2IP192 168 149 137 host1配置 spa
  • js中复选框checkbox如何判定为被选中

    在学习的过程中触及到html分离式开发的内容 xff0c 其中复选框checkbox判定选中对于ajax传数据来说十分重要 xff0c 故将该篇文章转发留存 xff0c 方便以后查阅 要想判定复选框选中状态可以使用 xff1a 34 tes
  • springboot使用ssh公钥连接mysql(含账号密码连接)

    引言 在项目开发过程中 xff0c 遇到了连接数据库时需要使用ssh公钥的情况 在本地使用navicat可以直接通过可视化界面去进行ssh的连接 xff0c 但是在java中无法直接去进行连接 后来经过查询资料 xff0c 发现必须要在ja
  • Spring boot使用Maven Profile配合Spring Profile进行多环境配置和打包

    引言 参考文章 xff1a https cloud tencent com developer article 1769239 最近在项目中有遇见maven的多环境配置 xff0c 由于以前没有接触过 xff0c 所以查阅资料后记录为笔记
  • idea中使用maven进行多模块打包部署时jar包中无依赖的问题(示例:jar包才5k,且无法正常启动)

    引言 模块关系 xff1a A xff1a 代表父工程B xff1a 代表WEB模块 xff0c 主加载类所在模块S xff1a 业务处理模块D xff1a 数据处理模块E xff1a 基础模块 依赖关系如下 xff1a D依赖ES依赖EB
  • 汇总maven配置失败问题

    1 maven的下载路径里面有中文或者符号 每个路径都要用英文 xff01 xff01 xff01 2 在系统环境中新建 变量名 xff1a MAVEN HOME 变量值 xff1a D Program Files apache maven
  • emq中测试连接8083成功,连接8084失败

    今天在emq中测试连接 xff0c 发现不加密的8083可以连上 xff0c 但加密的8084却连接不上 xff0c 搞半天才发现还有ssl文件未配置 经过找资料得知 xff0c 我是因为服务器上的emqx未配置 1 在域名配置ssl证书后
  • 〖小狼毫〗小狼毫初次使用配置

    这次配置是在中文全拼方式下进行的 xff0c 其他的输入方式可参考一下 xff1b 配置相关资料均来自于网络 一 这次配置实现的功能 英文单词输入日期时间类型输入 xff08 包括年 月 日 星期 几点几分 xff09 表情及颜文字输入 x
  • Windows10桌面美化——打造简洁高效美观桌面

    这是小编的Windows桌面 xff1a 常用的软件都固定在了任务栏上且进行了居中美化 所有的软件都固定在了开始屏幕上 我们的思路是 xff1a 把所有软件全部放在开始屏幕中 xff0c 需要使用时 按下键盘上的 window徽标键 打开
  • 学习笔记:医学图像配准简介—附voxelmorph模型

    VoxelMorph官方代码 xff1a GitHub voxelmorph voxelmorph Unsupervised Learning for Image Registrationb 本文参考了博客 xff1a 基于深度学习的单模医
  • 解决Powershell前面没有base,无法激活虚拟环境问题

    解决Powershell前面没有base xff0c 无法激活虚拟环境问题 1 Windows Powershell与CMD区别 在Win10系统下 xff0c 按住Shift然后点击鼠标右键进入Windows Powershell xff
  • LAMT配置

    tomcat项目部署 java环境安装 span class token punctuation span root 64 host span class token punctuation span span class token co
  • atc: invalid option -- ‘-‘

    最近在参加CANN训练营第三期 xff0c 在做应用课后作业的时候使用atc转化工具出现了如下错误 xff0c 百思不得其解 xff0c 看错误的原因是输入的atc命令都是合法的 xff0c 再看最后一行发现atc在 usr share g
  • 使用Anaconda安装Pychorm和torch ,cmd窗口进入python环境没有问题,但是在Pychorm中编写代码报错找不到torch包

    遇到这种问题很可能的原因是在Pychorm中创建工程时候选择的python解释器选择错误 xff0c 如果自己的电脑上之前就有安装的python IDE xff0c 但是没有安装torch包 xff0c 当选择这个解释器的时候就会出现找不到
  • MindSpore实现手写数字识别代码

    MindSpore是华为自研的一套AI框架 xff0c 最佳匹配昇腾处理器 xff0c 最大程度地发挥硬件能力 作为AI入门的LeNet手写字体识别网络 xff0c 网络大小和数据集都不大 xff0c 可以在CPU上面进行训练和推理 下面是
  • ATC模型转换

    xff08 1 xff09 在使用昇腾硬件 xff08 例如Atlas200 xff09 进行模型推理的时候 xff0c 首先需要使用ATC工具将原始模型转换为适合昇腾硬件的模型 xff08 om xff09 xff0c 支持原始框架类型为