Introduction

The 2021 NFL Health & Safety, Helmet Assignment competition is part of a collaborative effort between National Football League (NFL) and Amazon Web Services (AWS) to assist in the development of the best sports injury surveillance and mitigation program in the world. We are trying to assign the correct player “label” on helmets in NFL game footage. Labels consist of a value H (for home team) and V (for visiting team) followed by the player’s jersey number. Player labels are provided within the Next Gen Stats (NGS) tracking data for each play. A perfect submission would correctly identify the helmet box for every helmet in every frame of video and assign that helmet the correct player label.

Solution

The approach is fairly simple:

  1. Step through each frame in a video and apply the deepsort algorithm. This clusters helmets across frames when it is the same player/helmet.
  2. Group by each of these deepsort clusters - and pick the most common label for that cluster. Then override all of the predictions for that helmet to the same player.

References

NFL Helmet Assignment - Getting Started Guide

Helper Code + Helmet Mapping + Deepsort

NFL helmet with Yolov5-deepsort starter