Skip to content

My first exploration of the Tensor Flow Object Detection API, applied to ice hockey footage of the Vancouver Canucks.

Notifications You must be signed in to change notification settings

jamietownsend/CanucksObjectDetection

 
 

Repository files navigation

CanucksObjectDetection

I used the TensorFlow Object Detection API to apply a pre-trained model on ice hockey footage from my favorite team, the Vancouver Canucks. I wanted to see if the pre-trained 'person' class based on the COCO dataset could be used to detect the positions of hockey players on the ice.

I did this by modifying the Object Detection Tutorial jupyter notebook to only output overlays of the 'person' class ('skis' was the second most commonly identified class) and also lower the detection threshold from 0.5 to 0.25.

Inference from detection on still images was mixed (see example outputs). Detection on still images mixed

However, inference from detection on videos worked pretty well at lower thresholds. There's still noise where multiple 'persons' are detected for the same player, and certain poses (e.g., the goalie going into butterly) make the model fail, but certainly for the whole scene players are fairly detectable. Detection on videos pretty good

About

My first exploration of the Tensor Flow Object Detection API, applied to ice hockey footage of the Vancouver Canucks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 100.0%