this is my first project to study AI and try to detect safety helmets
if you want to use it, please understand that the dataset I use comes from the internet and is for educational purposes only and not for business. I'm not responsible for copyright infringement if you use it in the future, thank you.
To improve accuracy, you need to add more data objects to each class to train to produce a better model.
if you want to detect object using file while running main.py use or change it with following command in main.py using your .mp4 files: cap=cv2.VideoCapture("helmet.mp4")
if you want to detect object using facecam or realtime while running main.py use or change it with following command in main.py file: cap=cv2.VideoCapture(0)cap=cv2.VideoCapture(0)
Happy Hacking :)