Skip to content

Casassarnau/semi-supervised-learning-oxford-IIIT-pet

Repository files navigation

Exploring Transfer Learning and Semi-Supervised Learning: Classifying Cats and Dogs

This is the repository for the project of DD2424 Deep Learning in Data Science course at KTH, implemented by Silvia Arellano García, Arnau Casas Saez, and Nora Dunder.

Contents

In the files, you can find:

  • cat_n_dogs_classifier.ipynb: Basic binary classifier.
  • breeds_classifier.ipynb: Classifier of 37 breeds of cats and dogs, using a network that can be modified (using or not Batch Normalization, Learning Rate Decay, more than 1 layer...).
  • semi-supervised_learning_balanced_predictions.ipynb: Method explained in section 4.3.3 of the report.
  • semi-supervised_learning_threshold_predictions.ipynb: Methods explained in sections 4.3.1 and 4.3.2 of the report.
  • semi_supervised_learning_autoencoder.ipynb: Method explained in section 4.3.4 of the report.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published