/effective-catvdogs-classifier

A model for recognizing cats versus dogs.

Primary LanguageJupyter Notebook

Effective-CatsvDogs-Classifier

A model for recognizing cats versus dogs.

Training a state-of-the-art model for recognizing cats versus dogs.

  1. A dataset called the Oxford-IIIT Pet Dataset that contains 7,349 images of cats and dogs from 37 different breeds will be downloaded from the fast.ai datasets collection to the GPU server you are using, and will then be extracted.
  2. A pretrained model that has already been trained on 1.3 million images, using a competition-winning model will be downloaded from the internet.
  3. The pretrained model will be fine-tuned using the latest advances in transfer learning, to create a model that is specially customized for recognizing dogs and cats.