deepstream-python yolov5

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

Requirements

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.