# data.yaml
path: .
train: images
val: images
nc: 2
names:
  0: helmet
  1: head