NFL Health & Safety - Helmet Assignment: Segment and label helmets in video footage

Project setup

One can setup the project with the following command (preferably in a virtual environment):

make setup_project

Helmet detection notebook

The notebook yolov5_helmet_detection.ipynb displays an application of YOLOv5 to helmet detection. Running the whole notebook will generate a video clip with inference results obtained on one of the test videos. This video clip can be viewed in the last cell of the notebook.

Be careful to specify well where you store Kaggle's data for this challenge (in the second code cell of the notebook).

Pipeline

The goal of this competition is to assign specific players to each helmet.

In order to do so, the targeted pipeline of this project is the following:

1. Helmet detection with an object detection algorithm such as YOLOv5. There may be issues with sideline players' helmets.
2. Helmet tracking so that several different instances do not switch with each other (SORT algorithm?). As there may be many players in a very small area, this can be challenging.
3. Use the provided tracking data to identify players.