Create safety.yaml

safety_analyzer/safety.yaml

@@ -0,0 +1,14 @@
+# Paths to training/validation data
+train: ./safety_dataset/images/train
+val: ./safety_dataset/images/val
+
+# Number of classes
+nc: 5
+
+# Class names (match with your label IDs)
+names: 
+  0: no_helmet
+  1: no_harness
+  2: unsafe_posture
+  3: unsafe_zone
+  4: improper_tool_use