YOLOv5-cls-TensorRT-Cplusplus This is a TensorRT project based on yolov5-cls, using C++ 演示视频:https://www.bilibili.com/video/BV1Th4y1x7b8/?spm_id_from=333.999.0.0&vd_source=4cd5ac8dda02d0b3152cd9b05f7e4006 1.修改CmakeLists.txt中的cuda和tensorrt路径 2.修改main.py中的labels,onnx_file和ori_img 3.修改src/yolov5_cls.cpp中的IMAGE_WIDTH、IMAGE_HEIGHT、OUTPUT_SIZE,分别是输入图像的宽和高还有分类的类别数 4.将转好的onnx模型放入到models文件夹中 5.开始运行 第一次运行会把onnx模型转为.engine模型,时间稍微长一些 之后运行就直接调用.engine模型进行推理