zhouyuchong / retinaface-deepstream-python

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RetinaFace with Deepstream 6.0

Simple Deepstream pipeline uses Retinaface as inference. Demostrate how to parse raw-output and draw landmarks on osd. preview

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/mrscarletzhou/retinaface-deepstream-python.git

Usage

modify tensorRT engine network size in retina-engine-config file

python3 main.py {VideoPath}

References

About

License:Other


Languages

Language:Python 74.8%Language:C++ 18.8%Language:Makefile 6.4%