支持框架类型
- 支持的原始框架类型、输入数据类型和模型文件如表1所示。
表1 支持框架类型 原始框架类型
输入数据类型
模型文件说明
Caffe
- FP32
- FP16:通过设置入参--input_fp16_nodes实现。
- UINT8:通过配置数据预处理实现。说明:
输入数据最大支持四维,转维算子(reshape、expanddim等)不能输出五维。
- 模型文件:xxx.prototxt
- 权重文件:xxx.caffemodel
其中,模型文件和权重文件的op_name、op_type必须保持名称一致(包括大小写)。
TensorFlow
- FP16
- FP32
- UINT8
- INT32
- INT64
- BOOL说明:
不支持输出数据类型为INT64,需要用户自行将INT64的数据类型修改为INT32类型。
模型文件:xxx.pb
只支持FrozenGraphDef格式的.pb模型转换。
ONNX
- FP32
- FP16:通过设置入参--input_fp16_nodes实现。
- UINT8:通过配置数据预处理实现。
模型文件:xxx.onnx
MindSpore
- FP32
- UINT8:通过配置数据预处理实现。
模型文件:xxx.air
父主题: 模型转换