Skip to main content
 首页 » 编程设计

python之Tensorflow 1.6 中的 SSD_MOBILENET V1 到 TensorRT

2025年01月19日35leader

各位,

我有一台安装了 tensorflow 1.6 和 jetpack 3.2 (tensorrt 3.0) 的 Jetson TX2。我也安装了 UFF。 如何将 ssd_mobilenet_v1 卡住图从 tensorflow 转换为等效的 tensorRT?

谢谢,

请您参考如下方法:

访问 github 存储库:https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification 按照说明进行操作,它应该可以帮助您。 您应该将卡住图转换为计划(TensorRT 引擎等价物),然后将其序列化,在执行推理时反序列化 TensorRT 引擎并执行推理。