基于tensorrt的车道线检测lane_detection,实现从pth模型转engine模型,在通过tensorrt加载engine全过程

林君博
2023-12-01

问题以及解决方案

问题简述: 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 
 类似资料: