模型转换
- 进入模型转换功能。
在MindStudio上方菜单栏中选择
或工具栏单击,进入模型转换。
- 在“Model Information”页面下,完成各参数配置,如图1所示,填写完成,单击“Next”进入下一步。
- Model File:选择已上传的预训练模型(.pb)文件。
- Target Soc Version:选择处理器型号,本次样例以“Ascend310”为例。
- Output Path:转换后的om模型输出目录,本次样例选择工程目录下“models”文件夹。
- Input Nodes:将“data_1”的“shape”修改为“1,260,260,3”,原来的“-1”表示为动态batch,本次样例暂时只涉及单张图片,不涉及动态batch。
- 在“Data Pre-Processing”页面中,选择启用“Data Preprocessing”功能并在出现的功能选项中,选择启用“Load Aipp Configuration”,填写结果如图2所示,单击“Next”进入下一步配置。
- Image Pre-procesing Mode:Static。
- Aipp Configuration File:选择加载AIPP配置文件,文件位置在“models”目录下。
- 在“Advanced Options Preview”页面中,确认无误后单击“Finish”,开始转换。
- 如返回以下信息,则模型转换成功并可在配置的输出目录下找到.om文件,如图3所示。
Model converted successfully.
父主题: 应用开发(MindSDK)