更新时间:2022-08-05 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)开发的自定义算子。
父主题: 管理算法模型