我在将yolo.weight模型转换为tensorRT时遇到了一些问题。那么,有可能在Jetson中运行一个yolo模型并将其优化为TensorRT吗?会有同样的检测速度吗?(无论如何,训练不会在杰特森完成(。
或者除了TensorRT还有其他建议吗?
是的,在Jetson中运行YOLO模型是可能的,而无需使用TensorRT对其进行优化。TensorRT仅优化模型的推理时间。
您可以尝试将其转换为TF-Lite格式,然后从那里开始工作,但您可能需要自己处理大部分后端操作。
此外,对于这两种方法,训练都是在PC上进行的,而不是在边缘设备上。
你可以在这些链接中阅读更多关于他们的文档
Tensorflow RT Github
Tensorflow Lite