支持框架类型

模型转换和调优支持的原始框架类型、输入数据类型和模型文件如表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