CoolFace
Apppublic

Crypt0xDev/ppe-helmet-detection-api

sourceHugging Facemitupdated 4mo agoView on Hugging Face
0likes
App README

PPE Helmet Detection API

API de detección de cascos de seguridad usando YOLOv8 y FastAPI.

Inicio rápido

bash
pip install -r requirements.txt
cp .env.example .env
python app.py

Documentación interactiva: http://localhost:7860/docs

Endpoints

MétodoRutaDescripción
GET/Health check
POST/api/v1/detect/Detección → JSON con estadísticas
POST/api/v1/detect/imageDetección → imagen anotada (JPEG)
GET/api/v1/camera/snapshotCaptura desde cámara en tiempo real
POST/api/v1/alert/whatsappEnvío de alerta por WhatsApp

Variables de entorno

VariableDescripciónDefault
MODEL_PATHRuta al modelo YOLOv8models/best.pt
CONFIDENCE_THRESHOLDUmbral de confianza (0.0–1.0)0.35
IOU_THRESHOLDUmbral IoU para NMS (0.0–1.0)0.45
HOSTHost del servidor0.0.0.0
PORTPuerto del servidor7860
ALLOWED_ORIGINSOrígenes CORS (separados por coma)*
WHATSAPP_API_KEYAPI Key de CallMeBot (opcional)—
WHATSAPP_PHONENúmero de destino WhatsApp (opcional)—

Despliegue en Hugging Face Spaces

  1. 1.Ve a tu Space → Settings → Repository secrets
  2. 2.Agrega las variables de la tabla anterior con sus valores reales

Stack

YOLOv8 · FastAPI · OpenCV · Python 3.10+

Licencia

MIT — Universidad Nacional de San Martín (UNSM)