Download YOLO weights file: weights_cfg_names Put them in the root directory or change the path in yolo_helmet.py file to these files.
Required Libraries to run both files: - python - OpenCv - time Install other missing libraries.
Put your required images in the input folder and run the yolo_helmet.py.
Images with helmets will get marked and placed inside output folder. No of detected helmet will be on the terminal.