基于tensorrt的车道线检测lane_detection,实现从pth模型转engine模型,在通过tensorrt加载engine全过程
-
- 问题以及解决方案
- 如何运行代码
- 学习参考
- 代码实例讲解
- 将低版本tensorrt升高到高版本,以及最终效果
问题以及解决方案
问题简述: AttributeError: ‘tensorrt.tensorrt.Builder’ object has no
attribute ‘build_cuda_engine’ 原因:
tensorrt版本大于等于7时,不再使用build_cuda_engine
解决办法:
print('[info] Building an engine...')
if TRT_VERSION_MAJOR >= 7