- 本项目是本人为学员授课《C++ TensorRT高性能部署》的项目演示
- Ubuntu GPU
- TensorRT 8.4
- C++ 14
- 下载权重
yolov5s_person.onnx
,解压到weights
,下载地址 cmake . -B build && cmake --build build
- 在你的机器上构建engine:
./build/build [onnx_file_path] [calib_dir] [calib_list_file]
- 运行
./build/runtime_thread ./weights/yolov5.engine {视频文件} 2 50 0 2000000
- 使用VLC等客户端读取视频流:
rtmp://localhost:1936/live/mystream