This is a simple app build on the top of deepstream-test1 using custom tensorrt yolov5.
- Deepstream 6.0
- GStreamer 1.14.5
- Cuda 11.4+
- NVIDIA driver 470.63.01+
- TensorRT 8+
Follow deepstream official doc to install dependencies.
Deepstream docker is more recommended.
Please refer to this repo for pretrained models and serialized TensorRT engine.
git clone https://github.com/zhouyuchong/yolov5-deepstream-python
cd yolov4-deepstream-python
make nvdsinfer_custom_impl_Yolo
check all paths in deepstream_yolov5_config.txt and main.py. make sure they are correct.
python3 main.py {VideoPath}
this app only supports h264 format file.