face-recognition-deepstream

KO | EN

gitlite — search

#python #hacktoberfest #react #llm #golang #rust #deep-learning #linux #windows #vue #security #ios #openai #open-source

face-recognition-deepstream

(★ 85)

Deepstream app use retinaface and arcface for face recognition.

File Explorer

Download Latest Version (.zip)

- config_arcface.txt
- config_pipeline.toml
- config_retinaface.txt
- config_tracker_NvDCF_perf.yml
- config_tracker_perf.txt
- config_yolov8n_face.txt
- - readme
- - readme
- - Chandler.npy
  - Chandler.png
  - Monica.npy
  - Monica.png
  - Phoebe.npy
  - Phoebe.png
  - Rachel.npy
  - Rachel.png
  - Ross.npy
  - Ross.png
- - friends_s1e1_cut.mp4
- - readme
- - labels.txt
- - labels.txt
- bus_call.py
- gen_feature.py
- parser_cfg.py
- probe.py
.gitignore
.gitmodules
docker-compose.yml
gst-nvinfer-custom
LICENSE
main.py
README.es-ES.md
README.md
requirements.txt

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Face recognition with Deepstream
This is a face detection and recognition demo pipeline build on Deepstream.

## Requirements
+ Deepstream 6.0+
+ GStreamer 1.14.5+
+ Cuda 11.4+
+ NVIDIA driver 470.63.01+
+ TensorRT 8+
+ Python 3.6+
+ Opencv

Follow [deepstream](https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_Quickstart.html#dgpu-setup-for-ubuntu) official doc to install dependencies.

Deepstream docker is more recommended.

## Models
+ [yolov8-face](https://github.com/derronqi/yolov8-face)
+ [retinaface](https://github.com/biubug6/Pytorch_Retinaface)
+ [arcface](https://github.com/deepinsight/insightface/releases/tag/v0.7)

## Alignment

[gst-nvinfer-custom](https://github.com/zhouyuchong/gst-nvinfer-custom)

## Usage
### 1 - prepare data
+ put the feature file(.npy format) to `data/known_faces`
+ or put face images to `data/unknown_faces` and run `python3 utils/gen_feature.py`
### 2 - compile 
+ `gst-nvinfer-custom` : follow [README](https://github.com/zhouyuchong/gst-nvinfer-custom)
+ `nvdsinfer_customparser` for detector post-process
### 3 - run 
```
python3 main.py
```

## References
+ [NVIDIA-AI-IOT/deepstream_python_apps](https://github.com/NVIDIA-AI-IOT/deepstream_python_apps)
+ [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface)
+ [yolov8-face](https://github.com/derronqi/yolov8-face)
+ [arcface](https://github.com/deepinsight/insightface)

face-recognition-deepstream

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

face-recognition-deepstream

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Download README