更新时间:2022-08-18 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)开发的自定义算子。
分享:

    相关文档

    相关产品