thalostech2025/thalos-construction-ppe-v1

Thalos Construction PPE Detection – v1.0 (Roboflow → HuggingFace export)

This repository contains the Construction PPE Detection model used in the Thalos Safety Intelligence platform.

It identifies construction-specific PPE compliance and related hazards, including:

• hard hats (helmets)
• safety glasses
• high-visibility vests
• gloves
• missing PPE on workers
• partially worn or incorrectly used PPE
• workers present on active construction sites without adequate PPE

This model was originally trained in Roboflow (Object Detection Large / YOLOv8-derived)
and exported as a PyTorch .pt file.
It is now hosted on HuggingFace for independent, cost-controlled inference inside the Thalos YOLO service.

---

🚀 Usage (Python)

from ultralytics import YOLO

model = YOLO("thalostech2025/thalos-construction-ppe-v1/construction_ppe_weights.pt")

results = model("example.jpg")
results.show()


⸻

🔗 Load From URL (Used by the Thalos YOLO Service)

import torch
import requests
from io import BytesIO

HF_URL = "https://huggingface.co/thalostech2025/thalos-construction-ppe-v1/resolve/main/construction_ppe_weights.pt"

response = requests.get(HF_URL)
model = torch.load(BytesIO(response.content), map_location="cpu")


⸻

🏷️ Labels / Classes

Based on the v1.0 Roboflow dataset, the following classes are included:

hard_hat
no_hard_hat
safety_glasses
no_safety_glasses
hi_vis_vest
no_hi_vis_vest
gloves
no_gloves

(Class names may vary slightly depending on the finalized dataset schema.)

⸻

📄 License

This model is distributed under the AGPL-3.0 license, consistent with Roboflow export requirements.

⸻

🔒 Important Notes
    •	This repository contains only the model weights, not training data.
    •	Thalos uses this model via server-side inference in the YOLO microservice.
    •	Future versions (v2.0+) may unify Construction PPE and Industrial PPE into a shared multi-industry PPE model.
    •	This migration removes reliance on Roboflow cloud inference to control cost and eliminate external runtime dependencies.

⸻

👤 Maintainer

Thalos Tech (2025)
Safety Intelligence & Hazard Detection Platform