1.模型转换工具介绍
在进行模型转换前,请务必查看如下约束要求:
- 支持原始框架类型为Caffe、TensorFlow、MindSpore、ONNX的模型转换:
- 当原始框架类型为Caffe、MindSpore、ONNX时,输入数据类型为FP32、FP16(通过设置入参–input_fp16_nodes实现,MindSpore框架不支持该参数)、UINT8(通过配置数据预处理–insert_op_conf实现);
- 当原始框架类型为TensorFlow时,输入数据类型为FP16、FP32、UINT8、INT32、INT64、BOOL。当原始框架类型为Caffe时,模型文件(.prototxt)和权重文件(.caffemodel)的op name、op type必须保持名称一致(包括大小写)。
- 当原始框架类型为TensorFlow时,只支持FrozenGraphDef格式。
- 不支持动态shape的输入,例如:NHWC输入为[?,?,?,3]多个维度可任意指定数值。模型转换时需指定固定数值。
- 对于Caffe框架网络模型: 输入数据最大支持四维,转维算子(reshape、expanddim等)不能输出五维模型中的所有层算子除const算子外,输入和输出需要满足dim!=0。
只支持《算子规格说明》中的算子,并需满足算子限制条件。