昇腾社区首页
中文
注册

模型转换

  1. 进入模型转换功能。

    在MindStudio上方菜单栏中选择Ascend > Model Converter或工具栏单击,进入模型转换。

  2. “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。
    图1 Model Information页面
  3. “Data Pre-Processing”页面中,选择启用“Data Preprocessing”功能并在出现的功能选项中,选择启用“Load Aipp Configuration”,填写结果如图2所示,单击“Next”进入下一步配置。
    • Image Pre-procesing Mode:Static。
    • Aipp Configuration File:选择加载AIPP配置文件,文件位置在“models”目录下。
    图2 Data Pre-Processing页面
  4. “Advanced Options Preview”页面中,确认无误后单击“Finish”,开始转换。
  5. 如返回以下信息,则模型转换成功并可在配置的输出目录下找到.om文件,如图3所示。
    Model converted successfully.
    图3 模型转换成功