CoolFace
Apppublic

NadyaPutt/hardhat-detector

sourceHugging Faceupdated 3mo agoView on Hugging Face
0likes
App README

🦺 Hard Hat Detector

HMSI Mini Bootcamp — Sesi 4: Deploy ke Web App

Deteksi otomatis penggunaan helm/APD di area konstruksi menggunakan YOLOv8.

Demo

Upload foto konstruksi, pabrik, atau foto dengan orang — model akan mendeteksi:

  • —🟢 helmet — helm terdeteksi (aman)
  • —🔴 head — kepala tanpa helm (bahaya)
  • —🟣 person — orang

Stack

LayerTeknologi
ModelYOLOv8n (Transfer Learning)
BackendFastAPI + Python 3.11
FrontendReact 18 + TypeScript + Tailwind
DeployDocker → Hugging Face Spaces

Local Development

bash
# Clone repo
git clone https://huggingface.co/spaces/USERNAME/SPACE_NAME
cd SPACE_NAME

# Backend
pip install -r backend/requirements.txt
uvicorn backend.main:app --reload --port 8000

# Frontend (terminal baru)
cd frontend
npm install
npm run dev
# → buka http://localhost:5173

Deploy Ulang

bash
git add .
git commit -m "update"
git push
# HF Spaces auto-build dan deploy

Model Kustom (dari Sesi 3)

Copy best.pt dari hasil training ke root folder:

bash
cp /path/ke/best.pt ./best.pt

Lalu set di Dockerfile atau env var:

MODEL_PATH=best.pt