解决tensorrt加速ICudaEngine* build_engine(unsigned int, nvinfer1::IBuilder, nvinfer1::IBuilderConfig,_nvinfer1::icudaengine* build_engine(unsigned int,

作者：我家自动化 | 2024-04-14 20:25:35

踩

nvinfer1::icudaengine* build_engine(unsigned int, nvinfer1::ibuilder*, nvinf

原因是模型训练用的是yolov5x.pt，而我们用的命令是github上的 sudo ./yolov5 -s yolov5x.wts yolov5x.engine s，实际上命令应该为sudo ./yolov5 -s yolov5x.wts yolov5x.engine x。
即engine s应该改为engine x，如果训练用的m或者其他模型，也应该改成相应的参数！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/423956

解决tensorrt加速ICudaEngine* build_engine(unsigned int, nvinfer1::IBuilder*, nvinfer1::IBuilderConfig*,_nvinfer1::icudaengine* build_engine(unsigned int,

解决tensorrt加速ICudaEngine* build_engine(unsigned int, nvinfer1::IBuilder, nvinfer1::IBuilderConfig,_nvinfer1::icudaengine* build_engine(unsigned int,