哪些对象检测预训练模型可用并可与 TensorRT 转换?
Which object detection pre-trained models are available and convertible with TensorRT?
我正在研究使用 TensorRT 转换预训练的对象检测模型以在我的 NVIDIA Jetson TX2 上进行试用,但我发现的每个模型都有 TensorRT 尚不支持的层。到目前为止,我尝试了 SSD 与 MobileNet 和 Faster R-CNN,但它们都具有 TensorRT 不支持的诸如身份之类的操作,我找不到其他许多 TensorFlow 模型。
谢谢
此存储库包含您正在寻找的预训练模型,以及有关如何采用模型和构建 TensorRT 引擎的说明:https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification#download
我正在研究使用 TensorRT 转换预训练的对象检测模型以在我的 NVIDIA Jetson TX2 上进行试用,但我发现的每个模型都有 TensorRT 尚不支持的层。到目前为止,我尝试了 SSD 与 MobileNet 和 Faster R-CNN,但它们都具有 TensorRT 不支持的诸如身份之类的操作,我找不到其他许多 TensorFlow 模型。
谢谢
此存储库包含您正在寻找的预训练模型,以及有关如何采用模型和构建 TensorRT 引擎的说明:https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification#download