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

基于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

你可能感兴趣的:(深度学习,人工智能,deepstream,计算机视觉,pytorch)