Final Project of COMP576 Team 4 (Session 2)
Jinlin Li (jl288@rice.edu) Yang Zhou (yz164@rice.edu) Shinyou Zhang (xz93@rice.edu) Xiaoyuan Liu (xl120@rice.edu)
the ipynb file contains the codes for YOLOv7-based model training.
Please use Google Colab for training and download data from https://www.kaggle.com/datasets/whenamancodes/helmet-detection-at-work-for-safety to cloned YOLOv7 code directory. You should rename the data directory to "Dataset" for code to work.