Skip to content

quangdungluong/face-recognition-system

Repository files navigation

Face Recognition System

Face recognition pipeline powered by Triton Inference Server for flexible model deployment and inference.

Key Features

  • Model Flexibility - Easily swap detection and recognition models via Triton config
  • Efficient Inference - Optimized pipeline using NVIDIA Triton
  • Scalable Architecture - Distributed services for production workloads
  • Vector Storage - Fast similarity search with Qdrant
  • User-Friendly UI - Streamlit interface for face registration and management

Quick Start

# Start the system
docker-compose up -d
docker-compose -f docker-compose.qdrant.yaml up -d

# Launch face registration UI
streamlit run src/face_registration_app.py

# Launch application with webcam
python main.py --webcam

Services

Service Purpose Tech
Inference Server Model serving Triton
Vector DB Embedding storage Qdrant
Object Storage Image storage MinIO
Registration UI Face enrollment Streamlit

Swapping Models

  1. Add your model to the Triton model repository
  2. Update the model configuration in models/config.pbtxt
  3. Restart the Triton service

Face Registration

The Streamlit app provides an intuitive interface to:

  • Register new faces with name and attributes
  • Capture faces from webcam or upload images
  • View and manage existing face database
  • Test face recognition in real-time

License

MIT

About

Face recognition pipeline powered by Triton Inference Server

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages