AccelOnnxConverter 类

用于将模型转换为 AccelOnnx 风格的类。

继承

builtins.object

AccelOnnxConverter

构造函数

AccelOnnxConverter()

方法

convert_tf_model

将 TensorFlow 模型转换为 AccelOnnx 模型。

convert_tf_model

将 TensorFlow 模型转换为 AccelOnnx 模型。

static convert_tf_model(workspace, source_model, input_node, outputs_nodes, require_fpga_conversion=True)

参数

workspace

Workspace

必需

包含要检索的已注册模型的工作区对象

source_model

model 或 <xref:<xref:model id str>>

必需

要转换的已注册模型

input_node

str

必需

输入层的名称

outputs_nodes

<xref:<xref:str list>>

必需

图的输出节点的列表，其中每个节点都将代表网络的输出层

require_fpga_conversion

bool

默认值: True

指定是否需要 FPGA 来进行处理。 默认值为 True。

返回

等待转换并获取转换后的模型的对象

返回类型

<xref:ModelConvertOperation>