文档首页 > > 用户指南> 控制台开发技能> 管理算法模型>

模型输入目录规范

模型输入目录规范

分享
更新时间:2021/01/19 GMT+08:00

在本地开发模型后或在ModelArts开发并转换模型后,针对Ascend芯片,其模型输入目录需要满足一定的规范要求。华为HiLens当前对模型输入目录的要求如下:

  • 针对基于Caffe框架的模型,执行模型导入(转换)时,其输入目录需符合如下规范。
    |
    |---xxxx.caffemodel         模型参数文件,输入目录下有且只能有一个,必填。
    |---xxxx.prototxt           模型网络文件,输入目录下有且只能有一个,必填。
    |---insert_op_conf.cfg      插入算子配置文件,输入目录下有且只有一个,可选。
    |---plugin                  自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。
  • 针对基于TensorFlow框架的模型(“frozen_graph”“saved_model”格式),执行模型导入(转换)时,其输入目录需符合如下规范。
    “frozen_graph”格式
    |
    |---xxxx.pb                 模型网络文件,输入目录下有且只能有一个,必填。支持以frozen_graph或saved_model格式保存的模型。
    |---insert_op_conf.cfg      插入算子配置文件,输入目录下有且只有一个,可选。
    |---plugin                  自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。

    “saved_model”格式

    |
    |---saved_model.pb          模型网络文件,输入目录下有且只能有一个,必填。支持以frozen_graph或saved_model格式保存的模型。
    |---variables               固定子目录名称,包含模型的权重偏差等信息,必选
        |---variables.index     必选
        |---variables.data-00000-of-00001 必选
    |---insert_op_conf.cfg      插入算子配置文件,输入目录下有且只有一个,可选。
    |---plugin                  自定义算子目录,输入目录下有且只能有一个plugin文件夹,可选。仅支持基于TE(Tensor Engine)开发的自定义算子。
分享:

    相关文档

    相关产品

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问