【2023 · CANN训练营第一季】应用开发（初级）第三章——模型转换

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。
  只支持《算子规格说明》中的算子，并需满足算子限制条件。