# tensorrt-yolox

**Repository Path**: zhys513/tensorrt-yolox

## Basic Information

- **Project Name**: tensorrt-yolox
- **Description**: 基于TensorRT实现的YOLOX
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 2
- **Created**: 2021-12-30
- **Last Updated**: 2021-12-30

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 基于TensorRT实现的YOLOX

## 依赖

- CUDA
- TensorRT
- 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_yolox`。

## 执行

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

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

比如：

```shell
./tensorrt_yolox ../models/yolox_s.onnx ../video/test.mp4
```

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