【问题标题】:SSD_MOBILENET V1 to TensorRT in Tensorflow 1.6从 SSD_MOBILENET V1 到 Tensorflow 1.6 中的 TensorRT
【发布时间】: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


    【解决方案1】:

    访问 github 仓库:https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification 按照说明进行操作,它应该可以帮助您。 您应该将冻结的图转换为计划(等效于 TensorRT 引擎)然后对其进行序列化,同时执行推理时反序列化 TensorRT 引擎并执行推理。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-05-11
      相关资源
      最近更新 更多