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.

[参考译文] RTOS:如何将 TensorFlow MobileNet V1模型导入 TIDL 模型?

Guru**** 2538960 points


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

https://e2e.ti.com/support/processors-group/processors/f/processors-forum/794522/rtos-how-can-i-import-tensorflow-mobilenet-v1-model-to-tidl-model

工具/软件:TI-RTOS

您好!

我正在尝试按照 TIDLDeepLearningLibrary.pdf 第3.6.5节转换基于 TensorFlow-slim 的 MobileNet 模型(Ver1)。

我下载 了 MobileNet_v1_1.0_224.ckpt.data-00000-of 00001、mobilenet_v1_1.0_224.ckpt.index、MobileNet_v1_1.0_224.ckpt.meta、MobileNet_v1_1.0_224.PB。

然后我运行命令来优化 MobileNet_v1_1.0_224.PB。

但是、显示了这些警告消息。

----------------------------------------

警告:tensorflow:来自/usr/local/lib/python3.4/dist-packages/tensorflow/python/tools/strip_unused_lib.py:86:extract_sub_graph (来自 tensorflow.python.framework.graph_util_impl)已弃用、并将在将来的版本中删除。
更新说明:
请使用 tf.compat.v1.graph_util.extract_sub_graph
警告:tensorflow:来自/usr/local/lib/python3.4/dist-packages/tensorflow/python/tools/optimize_for_inference_lib.py:113:remove_training 节点(来自 tensorflow.python.framework.graph_util_impl)已弃用、并将在将来的版本中删除。
更新说明:
请使用 tf.compat.v1.graph_util.remove_training_nodes
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_1_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_2_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_3_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_4_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_5_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_6_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_7_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_8_depthise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_9_depthise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_10_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_11_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_12_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入
警告:tensorflow:未找到"MobilenetV1/MobilenetV1/Conv2d_13_depthwise/BatchNorm/FusedBatchNorm"的预期 Conv2D 输入

----------------------------------------

总之、我忽略了该消息、并使用结果 PB 文件转换为 TIDL 模型。

在我执行 tidl_model_import.out.exe D:\TI_TDA2x\ti_dl\testvecs\config\cimport\tidl_import_mobileNet1.txt 后、显示了此错误消息。

----------------------------------------

TF 模型文件:..\test\testvecs\config\tensorflow_models\MobileNet\MobileNet_1_224.PB
找不到请求的输入数据:MobilenetV1/MobilenetV1/Conv2d_0/Relu6!!

----------------------------------------

TIDL 导入工具不支持 Relu6层吗?

如果是、您能告诉我如何验证 TF MobileNet V1模型吗?

提前感谢。

柳时敏。

  • 请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。
    对于冻结图、请使用 TF 版本1.1。 我们还没有升级到最新版本的 TF。
  • 请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。

    谢谢你。

    在我重新安装 tensorflow 1.1.0版本后,它可以正常工作。

    此致

    柳时敏。