小编Sha*_*ran的帖子

在 python 上使用 TensorRT .engine 文件进行推理

我使用 Nvidia 的 Transfer Learning Toolkit (TLT) 进行训练,然后使用 tlt-converter 将 .etlt 模型转换为 .engine 文件。

我想在 python 中使用这个 .engine 文件进行推理。但由于我使用 TLT 进行训练,因此我没有任何冻结图或 pb 文件,而这些都是 TensorRT 推理教程所需要的。

我想知道在 .engine 文件上是否可以进行 python 推理。如果没有,支持哪些转换(UFF,ONNX)来实现这一点?

python computer-vision deep-learning tensorflow tensorrt

6
推荐指数
1
解决办法
1885
查看次数