在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 repo:https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification 按照说明操作，它应该能帮你解决问题。 您应该将冻结的图转换为计划（TensorRT Engine Equivalent），然后将其序列化；在执行推理时，您将反序列化TensorRT引擎并执行推理。在