Project in Machine Learning & Neural Networks, developed under the scope of the MSc course, Machine Learning & Content Analytics.
Helmet_classification_notebook is the master notebook for the project (data preprocessing, models, evaluation and etc.)
Cropped Images(2nd approach) is another notebook with a second solution with bounding box on heads.
Report_image_classification is the pdf report presenting our work.