Какие предварительно обученные модели обнаружения объектов доступны и могут быть преобразованы с помощью TensorRT?

Question

Я пытаюсь преобразовать предварительно обученную модель обнаружения объектов с помощью TensorRT, чтобы опробовать ее на моем NVIDIA Jetson TX2, но каждая найденная мной модель имеет слои, которые еще не поддерживаются TensorRT. До сих пор я пробовал SSD с MobileNet и Faster R-CNN, но у них обоих есть такие операции, как Identity, которые не поддерживаются TensorRT, и я не могу найти много других моделей TensorFlow.

Спасибо

Answer 1

В этом репозитории есть предварительно обученные модели, которые вы ищете, а также инструкции о том, как взять модель и построить движок TensorRT: https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification