mindspore_lite.FmkType

查看源文件
class mindspore_lite.FmkType[源代码]

当Converter时, FmkType 定义输入模型的框架类型。

目前,支持以下模型框架类型:

定义

说明

FmkType.TF

TensorFlow模型的框架类型,该模型使用.pb作为后缀。

FmkType.CAFFE

Caffe模型的框架类型,该模型使用.prototxt作为后缀。

FmkType.ONNX

ONNX模型的框架类型,该模型使用.onnx作为后缀。

FmkType.MINDIR

MindSpore模型的框架类型,该模型使用.mindir作为后缀。

FmkType.TFLITE

TensorFlow Lite模型的框架类型,该模型使用.tflite作为后缀。

FmkType.PYTORCH

PyTorch模型的框架类型,该模型使用.pt或.pth作为后缀。

样例:

>>> # Method 1: Import mindspore_lite package
>>> import mindspore_lite as mslite
>>> print(mslite.FmkType.TF)
FmkType.TF
>>> # Method 2: from mindspore_lite package import FmkType
>>> from mindspore_lite import FmkType
>>> print(FmkType.TF)
FmkType.TF