jcantero9/helmet-detection

Helmet Detection API

Backend FastAPI para detección de cascos de seguridad con YOLOv8s.

Endpoints

• —GET /api/health — Estado del servicio
• —POST /api/detect — Detectar en imagen subida (multipart/form-data)
• —POST /api/detect/frame — Detectar en frame de webcam (JSON base64)

Modelo

YOLOv8 Small entrenado con Hard Hat Workers Dataset (Kaggle).

• —mAP@0.5: 0.9328 | F1: 0.9022
• —Clases: helmet, no_helmet