# tensorrt-yolov6

**Repository Path**: xiaoyuerCV/tensorrt-yolov6

## Basic Information

- **Project Name**: tensorrt-yolov6
- **Description**: 用TensorRT实现的YOLOv6
- **Primary Language**: C++
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 11
- **Forks**: 3
- **Created**: 2022-07-01
- **Last Updated**: 2024-09-25

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 基于TensorRT实现的YOLOv6

**如果要用CV-CUDA做图像预处理，请切换到cvcuda分支**

## 依赖

- CUDA
- TensorRT(7/8)
- OpenCV

## 编译

首先在`CMakeLists.txt`文件中设置`CUDA`和`TensorRT`的安装路径：

```
set(CUDA_PATH /usr/local/cuda)
set(TENSORRT_PATH /path/to/TensorRT)
```

然后执行下面的命令进行编译：

```shell
mkdir build
cd build
cmake ..
make
```

编译成功后会生成可执行文件`tensorrt_yolov6`。

## 执行

执行下面的命令启动程序:

```shell
./tensorrt_yolov6 </path/to/onnx/model> </path/to/video>
```

比如：

```shell
./tensorrt_yolov6 ../models/yolov6_s.onnx ../video/test.mp4
```

程序默认会把检测结果保存到`result.mp4`文件中。

**扫描下面二维码关注我的公众号【DeepDriving】，我会持续分享计算机视觉、机器学习、深度学习、无人驾驶等领域的文章**
![](https://tuchuang-1312078823.cos.ap-guangzhou.myqcloud.com/markdown/gongzhonghao.jpg)