器件型号:PROCESSOR-SDK-DRA8X-TDA4X
您好,香榭丽舍
第一。 我正在使用 TIDL 体验 TDA4。
2。 在此文件中、需要在 Ubuntu 18上安装 tensorFlow 才能使用此文件: optimize_for_inference.py
3号。 请参阅以下部分。
我的问题是:
我们需要安装哪个 TensorFlow 版本?
需要选择哪个 Python 版本(2或3)?
请发表评论。
谢谢。
BR RIO
文件:///opt/psdk_rtos_auto_j7_06_02_00_21_original/tidl_j7_01_01_00_10/ti_dl/docs/user_guide_html/md_tidl_user_model_deployment.html。#环境设置
从此处下载包含已训练模型的 tarball。
您需要从 tarball 中提取 MobileNet_v2_1.0_224_Frozen.PB 并将其放入 ti_dl/test/testvecs/models/public/tensorflow/MobileNet_v2目录中。
下载的 tensorflow 模型只有在针对推理进行优化后才能导入。 运行 optimize_for_inference.py (随 tensorflow 安装分发)以创建优化的模型文件。
用户@Uubuntu-PC$ python optimize_for_inference.py \
--input=${tidl_install_path}/tidl/test/testvecs/models/public/tensorflow/MobileNet_v2/MobileNet_v2_1.0_224_Frozen.pb \
--output=${tidl_install_path}/ti_dl/test/testvecs/models/public/tensorflow/MobileNet_v2/MobileNet_v2_1.0_224_final.pb \
--input_names="input"\
--output_names="MobilenetV2/Predictions/Softmax"