data.yaml

train: ./dataset/images/train
val: ./dataset/images/val
test: ./dataset/images/test

nc: 3
names: ['no_helmet', 'unsafe_distance', 'unauthorized_area']