Skip to content

zhouyuchong/retinaface-deepstream-python

Repository files navigation

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

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published