【发布时间】:2018-05-01 08:46:27
【问题描述】:
伙计们,
我有一个安装了 tensorflow 1.6 和 jetpack 3.2 (tensorrt 3.0) 的 Jetson TX2。我也安装了 UFF。 如何将 ssd_mobilenet_v1 冻结图从 tensorflow 转换为等效的 tensorRT?
谢谢,
【问题讨论】:
标签: python tensorflow models object-detection tensorrt
伙计们,
我有一个安装了 tensorflow 1.6 和 jetpack 3.2 (tensorrt 3.0) 的 Jetson TX2。我也安装了 UFF。 如何将 ssd_mobilenet_v1 冻结图从 tensorflow 转换为等效的 tensorRT?
谢谢,
【问题讨论】:
标签: python tensorflow models object-detection tensorrt
访问 github 仓库:https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification 按照说明进行操作,它应该可以帮助您。 您应该将冻结的图转换为计划(等效于 TensorRT 引擎)然后对其进行序列化,同时执行推理时反序列化 TensorRT 引擎并执行推理。
【讨论】: