Tensorflow 1.6中的SSD_MOBILENET V1到TensorRT

Question

伙计们，

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

谢谢，

Answer 1

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