work/data.yaml

# Ultralytics dataset config for the PPE detection set (10 classes).
#
# NOTE: there is intentionally no `path:` key. Ultralytics resolves a relative
# `path:` against its global datasets_dir (`yolo settings`), which differs on
# every machine. With `path:` omitted, it instead resolves train/val/test
# relative to THIS file's directory (work/) — so the same config works locally,
# in Colab, and in CI, as long as the dataset sits next to it:
#
#   work/train/  work/valid/  work/test/   (git-ignored — see work/README.md)
#
# Fetch the data with:  uv run python scripts/download_data.py

train: train
val: valid
test: test

nc: 10
names:
  - Hardhat
  - Mask
  - NO-Hardhat
  - NO-Mask
  - NO-Safety Vest
  - Person
  - Safety Cone
  - Safety Vest
  - machinery
  - vehicle