---
license: cc-by-4.0
library_name: ultralytics
tags:
- object-detection
- ppe
- construction-safety
- yolov8
- ml-intern
---

# yolov8s-ppe-construction-v2

Improved PPE Compliance Detection Model for Construction Sites (v2)

## Classes (18)
- 0: person
- 1: helmet
- 2: vest
- 3: mask
- 4: gloves
- 5: safety_shoe
- 6: goggles
- 7: no_helmet
- 8: no_mask
- 9: no_vest
- 10: head
- 11: barricade
- 12: dumpster
- 13: excavators
- 14: safety_net
- 15: dump_truck
- 16: truck
- 17: wheel_loader

## Usage
```python
from ultralytics import YOLO
model = YOLO("hf://baskarmother/yolov8s-ppe-construction-v2/best.pt")
results = model.predict("image.jpg")
```

## Training Details
- Base Model: YOLOv8s
- Epochs: 150
- Image Size: 640x640
- Batch Size: 16

<!-- ml-intern-provenance -->
## Generated by ML Intern

This model repository was generated by [ML Intern](https://github.com/huggingface/ml-intern), an agent for machine learning research and development on the Hugging Face Hub.

- Try ML Intern: https://smolagents-ml-intern.hf.space
- Source code: https://github.com/huggingface/ml-intern