zhouyuchong / yolov5-deepstream-python

yolov5-deepstream-python

deepstream-python yolov5

This is a simple app build on the top of deepstream-test1 using custom tensorrt yolov5.

中文

Requirements

• 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.

Pretrained

Please refer to this repo for pretrained models and serialized TensorRT engine.

Installation

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.

Usage

python3 main.py {VideoPath}

this app only supports h264 format file.