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 Engine等效），然后对其进行序列化，同时在进行推理时对TensorRT引擎进行反序列化并进行推理。