python - TensorFlow 1.6 中的 SSD_MOBILENET V1 到 TensorRT

Question

伙计们，

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

谢谢，

Answer 1

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