我正在考虑使用 TensorRT 转换一个预训练的对象检测模型，以便在我的 NVIDIA Jetson TX2 上进行尝试，但我发现的每个模型都有 TensorRT 尚不支持的层。到目前为止，我尝试了使用 MobileNet 和 Faster R-CNN 的 SSD，但它们都具有 TensorRT 不支持的 Identity 等操作，而且我找不到许多其他 TensorFlow 模型。
谢谢
问问题
576 次
1 回答
0
这个 repo 包含您正在寻找的预训练模型，以及如何获取模型和构建 TensorRT 引擎的说明：https ://github.com/NVIDIA-Jetson/tf_to_trt_image_classification#download
于 2018-09-26T07:12:19.813 回答