shining-love / YOLOv5-cls-TensorRT-Cplusplus

This is a TensorRTproject based on yolov5-cls, using C++

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模型进行推理