Face Recognition Attendance System

Camera ─► Face Detect (YOLOv8-face / MTCNN)
       ─► Align + Crop (112x112)
       ─► Embedding (ArcFace / FaceNet) → 512-d vector
       ─► Cosine search in FAISS / pgvector
            │
            ├─ match (sim > 0.6) → log attendance
            └─ no match           → "unknown" alert
       ─► FastAPI dashboard + DB

pip install insightface onnxruntime-gpu faiss-cpu opencv-python sqlalchemy

from insightface.app import FaceAnalysis
import cv2, numpy as np

app = FaceAnalysis(name="buffalo_l", providers=["CUDAExecutionProvider"])
app.prepare(ctx_id=0, det_size=(640, 640))

def embed(image_path):
    img   = cv2.imread(image_path)
    faces = app.get(img)
    if not faces: raise ValueError("no face")
    return faces[0].normed_embedding   # (512,) float32

# 3–5 ছবি average → robust embedding
emb = np.mean([embed(p) for p in glob("sadiq/*.jpg")], axis=0)
emb /= np.linalg.norm(emb)
save_to_db(user_id="EMP001", name="Sadiqul Islam", embedding=emb)

import faiss, numpy as np, cv2, time
from datetime import datetime

index   = faiss.IndexFlatIP(512)        # cosine = inner product (normalized)
ids, names = load_all_from_db(index)
THRESHOLD  = 0.5

cap = cv2.VideoCapture(0)
seen_today = set()

while True:
    ok, frame = cap.read()
    if not ok: break
    for f in app.get(frame):
        q = f.normed_embedding.reshape(1, -1).astype("float32")
        sim, idx = index.search(q, 1)
        if sim[0][0] > THRESHOLD:
            uid = ids[idx[0][0]]
            if uid not in seen_today:
                mark_attendance(uid, datetime.now())
                seen_today.add(uid)
                cv2.putText(frame, names[idx[0][0]],
                            (int(f.bbox[0]), int(f.bbox[1])-10),
                            cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0,255,0), 2)
    cv2.imshow("Attendance", frame)
    if cv2.waitKey(1) == 27: break

CREATE EXTENSION vector;

CREATE TABLE employees (
  id           TEXT PRIMARY KEY,
  name         TEXT NOT NULL,
  embedding    VECTOR(512) NOT NULL,
  enrolled_at  TIMESTAMPTZ DEFAULT NOW()
);

CREATE TABLE attendance (
  id           BIGSERIAL PRIMARY KEY,
  employee_id  TEXT REFERENCES employees(id),
  ts           TIMESTAMPTZ DEFAULT NOW(),
  camera_id    TEXT,
  confidence   REAL
);

CREATE INDEX ON employees USING ivfflat (embedding vector_cosine_ops);

@app.post("/enroll")        # multipart photo upload
@app.get ("/attendance")    # date range query
@app.get ("/today")         # live attendance feed
@app.ws  ("/stream")        # websocket — real-time camera tile