G4-Venesyc/bigbang-actors-recognition

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

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! 🚀