模型输入目录规范.md

模型输入目录规范

在本地开发模型后或在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）开发的自定义算子。