昇腾社区首页
中文
注册

支持框架类型

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