SSD_MOBILENET V1 в TensorRT в Tensorflow 1.6

Question

Люди,

У меня установлен Jetson TX2 с tenorflow 1.6 и jetpack 3.2 (тензоррт 3.0). Я также установил UFF. Как я могу преобразовать замороженный граф ssd_mobilenet_v1 из тензорного потока в тензорный эквивалент?

Спасибо

Answer 1

Посетите репозиторий Github: https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification Следуйте инструкциям, это должно помочь вам. Вам следует преобразовать замороженные графики в план (эквивалентный TensorRT Engine), а затем сериализовать его, выполняя вывод, десериализовав механизм TensorRT и выполняя вывод.