准备离线模型dump数据
操作步骤
以下操作请在MindStudio环境上操作。
- 创建应用工程。
- 单击欢迎界面的“New Project”或菜单栏图1所示,单击“Next”。
- CANN Version:CANN软件包版本,如未识别或想要更换使用的版本,可单击“Change”,在弹出界面中选择Ascend-cann-toolkit开发套件包的安装路径(注意需选择到版本号或者latest一级)。
- Samples:样例工程,选择“ACL ResNet-50(C++)”。
,弹出“New Project”窗口。设置好相关参数后,如 - 配置相关信息,如图2所示,单击“Finish”。
- Project name:工程名称,可自定义。
- Project location:工程目录,默认在“$HOME/MindstudioProjects”下创建。
- 单击欢迎界面的“New Project”或菜单栏图1所示,单击“Next”。
- 将准备TensorFlow模型npy数据文件中准备的pb模型通过ATC工具转换为om模型。
- 单击菜单栏图3 模型转换页面
打开模型转换页面,选择待转换的pb文件,设置输出路径等,单击“Next”。
- 进入“Data Pre-Processing”配置数据预处理,单击“Next”。
- 进入“Advanced Options Preview”高级选项配置,单击“Finish”,开始进行模型转换。
出现“Model converted successfully”则表示转换成功,在指定的输出目录下生成*.om文件。
- 单击菜单栏
- 将准备好的输入文件test.bin上传至相应目录下,如图4所示。
模型推理输入test.bin文件准备可参考模型推理输入bin文件准备。
- 修改脚本,指定om模型和输入文件。
编辑src目录下的sample_process.cpp文件,修改参考如下。
- 单击菜单栏 ,进入编译配置页面。
单击
新增并配置编译参数,图5为配置示例,单击“OK”。
- 在菜单栏单击“Build>Build CMake Project”进行编译。
编译完成后main文件会生成在当前工程目录的out目录下。
图6 main文件所在目录 - 单击菜单栏 ,进入运行配置页面。
- 单击“
> CMake Application”新增并配置应用工程运行参数,图7为配置示例。单击“OK”。
- dump配置。
- 单击菜单栏“OK”。
图8 选择om模型
,选择om模型,如下图所示,单击
- 进行dump配置,参考如下。
图9 配置页面
- 单击菜单栏“OK”。
- 单击
运行应用工程,生成的dump数据保存在当前工程的dump目录下。
图10 dump目录
父主题: 比对数据准备