This thread has been locked.

If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.

[参考译文] TDA4VM-Q1:TIDL:如何在 TIDL 中导入我的 Tensorflow 模型?

Guru**** 2536150 points


请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。

https://e2e.ti.com/support/processors-group/processors/f/processors-forum/1072088/tda4vm-q1-tidl-how-to-import-my-tensorflow-model-in-tidl

部件号:TDA4VM-Q1

您好,专家,  

我按照 TIDL 指南导入 mobilenetV2模型(从 TIDL Link 下载模型)。 工作正常。

当我使用 Tensorflow API 创建 mobileenetv2模型时。  同样的方法,我遇到了下面的问题。  主要代码如下:  

型号= MobileNetV2 (权重='imagenet')

model.save("./saved_model_Fredy")

太棒了! LS ./saved_model_Fredy

资产	kernel_metadata.pb saved_model.pb 变量 

然后,我使用 saved_demel.pb 运行以下命令。 谢谢。

~/.local/lib/python3.6/site-packages/tensorflow/python/tools/ && python3 optimize_for_inference.py \
--input=/home/fredy/startJacinto/sdks/ti-processor-sdk-rtos-j721e-evm-08_01_00_11/`ls /home/fredy/startJacinto/sdks/ti-processor-sdk-rtos-j721e-evm-08_01_00_11 | grep tidl_`/ti_dl/test/testvecs/models/public/tensorflow/tf_fredy/tf_fredy.pb \
--output=/home/fredy/startJacinto/sdks/ti-processor-sdk-rtos-j721e-evm-08_01_00_11/`ls /home/fredy/startJacinto/sdks/ti-processor-sdk-rtos-j721e-evm-08_01_00_11 | grep tidl_`/ti_dl/test/testvecs/models/public/tensorflow/tf_fredy/tf_fredy_final.pb \
--input_names="input" \
--output_names="MobilenetV2/Predictions/Softmax"
Traceback (most recent call last):
File "optimize_for_inference.py", line 165, in <module>
app.run(main=main, argv=[sys.argv[0]] + unparsed)
File "/home/fredy/.local/lib/python3.6/site-packages/absl/app.py", line 303, in run
_run_main(main, args)
File "/home/fredy/.local/lib/python3.6/site-packages/absl/app.py", line 251, in _run_main
sys.exit(main(argv))
File "optimize_for_inference.py", line 83, in main
input_graph_def.ParseFromString(data)
google.protobuf.message.DecodeError: Error parsing message with type 'tensorflow.GraphDef'
 

请帮助检查并给出您的意见。 谢谢。  

此致,  

张瑞迪