如何将张量流模型(例如* .pb格式)转换为armnn(* .armnn格式)

时间:2020-01-21 12:09:31

标签: tensorflow converters

我想将tensorflow模型转换为armnn。我试图从零开始为Linux构建armnn。 我尝试了此链接-> https://github.com/ARM-software/armnn/blob/branches/armnn_19_11/BuildGuideCrossCompilation.md,但无法构建armnn Converter对象文件。 谁能帮助我成功构建它。

谢谢

1 个答案:

答案 0 :(得分:0)

您可以将armnn Converter用于同一任务。转换器只能转换有限的图层 对于tenserflow支持,您可以转到下面的链接并浏览图层。 https://github.com/ARM-software/armnn/blob/branches/armnn_20_08/src/armnnTfParser/TensorFlowSupport.md

在“ URL”下面,您可以获取armnn转换器的源。您可以独立构建 https://github.com/ARM-software/armnn/tree/branches/armnn_20_08/src/armnnConverter

对于转化,您可以使用以下示例中的命令

ArmnnConverter -f tensorflow-binary -m {pb_filename} -i {input_layers} -o 
{output_layer} -s {input-size} -p {model_name_after_conversion}