gpt4 book ai didi

python - Tensorflow 1.6 中的 SSD_MOBILENET V1 到 TensorRT

转载 作者:太空宇宙 更新时间:2023-11-04 04:43:27 25 4
gpt4 key购买 nike

各位,

我有一台安装了 tensorflow 1.6 和 jetpack 3.2 (tensorrt 3.0) 的 Jetson TX2。我也安装了 UFF。如何将 ssd_mobilenet_v1 卡住图从 tensorflow 转换为等效的 tensorRT?

谢谢,

最佳答案

访问 github 存储库:https://github.com/NVIDIA-AI-IOT/tf_to_trt_image_classification按照说明进行操作,它应该可以帮助您。您应该将卡住图转换为计划(TensorRT 引擎等价物),然后将其序列化,在执行推理时反序列化 TensorRT 引擎并执行推理。

关于python - Tensorflow 1.6 中的 SSD_MOBILENET V1 到 TensorRT,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50113650/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com