该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

基于TensorRT实现的YOLOX

依赖

  • CUDA
  • TensorRT
  • OpenCV

编译

首先在CMakeLists.txt文件中设置CUDATensorRT的安装路径:

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

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

mkdir build
cd build
cmake ..
make

编译成功后会生成可执行文件tensorrt_yolox

执行

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

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

比如:

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

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

空文件

简介

基于TensorRT实现的YOLOX 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化