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

1.模型转换工具介绍

image.png

在进行模型转换前,请务必查看如下约束要求:

  • 支持原始框架类型为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。
    只支持《算子规格说明》中的算子,并需满足算子限制条件。

image.png

image.png

image.png

猜你喜欢

转载自blog.csdn.net/qq_45257495/article/details/130874996