Skip to content

Image classifier built to classify oval sports balls into Rugby Balls, Australian Rules Footballs or American Footballs. CNN trained using transfer learning from the base ResNet34 architecture

Notifications You must be signed in to change notification settings

andrewtwort/oval_sports_ball

Repository files navigation

Small web app deploying a CNN built with the fast.ai library

CNN built using the Fast AI library. Model specifications:

  • Overview: CNN used to classify oval sports balls into Rugby Balls, Australian Rules Footballs or American Footballs
  • Architecture: ResNet34 (with optimised hyperparameters)
  • Data: Training set conisted of 575 images downloaded from Google Images (143 in the validation set). Datasets excluded images that were not photographs of oval sports balls (e.g. hand drawn images, clip-art, GIFs)
  • Training: ~10 minutes on a GPU hosted on Colab; 94.4% accuracy
  • Supporting materials: Largely follows lesson 2 of the Fast AI Practical Deep Learning for Coders (v3) course
  • Further reading: More on prolate spheroids here
  • Note: Rugby Union and Rugby League balls treated as one class

Sample images used in CNN training

oval sports balls Sample images: Wikipedia

About

Image classifier built to classify oval sports balls into Rugby Balls, Australian Rules Footballs or American Footballs. CNN trained using transfer learning from the base ResNet34 architecture

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages