我正在研究使用TensorRT转换预训练对象检测模型,以便在我的NVIDIA Jetson TX2上进行试用,但我发现的每个模型都有TensorRT尚不支持的图层。到目前为止,我尝试使用MobileNet和更快的R-CNN SSD,但是他们都有TensorRT不支持的身份等操作,而且我找不到其他许多TensorFlow模型。
谢谢
答案 0 :(得分:0)
此存储库包含您正在寻找的经过预先训练的模型,以及有关如何建立模型和构建TensorRT引擎的说明:https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification#download