You can deploy this model and start earning money today!

G4-Venesyc/
bigbang-actors-recognition

Views

No views yet

Face Recognition Model - Hugging Face

Este repositorio contiene un modelo de reconocimiento facial basado en InsightFace para procesar videos e identificar rostros mediante embeddings preentrenados.

📌 Características

Utiliza InsightFace (buffalo_l) para la detección y reconocimiento facial.
Procesamiento optimizado con CUDA (si está disponible).
Maneja procesamiento en batches para mayor eficiencia.
Compara rostros detectados con embeddings de referencia almacenados en reference_embeddings.pkl.

🚀 Instalación

1️⃣ Clonar el repositorio

bash

git clone https://huggingface.co/G4-Venesyc/bigbang-actors-recognition
cd bigbang-actors-recognition

2️⃣ Instalar dependencias

pip install -r requirements.txt

📌 Requisitos clave:

opencv-python
insightface
onnxruntime-gpu (para uso en GPU) o onnxruntime (para CPU)

🛠 Uso del modelo

1️⃣ Preparar archivos de entrada

Coloca imágenes de los videos en la carpeta datasets/images/ dentro de subcarpetas (una por video).
Asegúrate de tener el archivo reference_embeddings.pkl con los embeddings de referencia.

2️⃣ Ejecutar el procesamiento

python face_video_processor.py

📌 Opcional: Si deseas cambiar el tamaño del batch, puedes modificar el parámetro batch_size en el script.

📂 Estructura del repositorio

/
│── face_video_processor.py  # Código principal
│── reference_embeddings.pkl # Embeddings de referencia
│── datasets/
│   ├── images/  # Carpeta con imágenes organizadas por video
│── outputs/  # Resultados en formato JSON
│── requirements.txt  # Dependencias necesarias
│── README.md  # Este archivo

📜 Referencias

InsightFace: https://github.com/deepinsight/insightface
Hugging Face Spaces: https://huggingface.co/spaces

¡Listo para usar! 🚀