Classification de races de chiens

Ce projet à pour but de créer un modèle de réseaux de neurones convolutifs à partir des données du Stanford Dogs Datasets qui contient des 20,580 images de 120 races de chiens du monde entier.

L’objectif est de proposer une web app pour obtenir la race d’une chien à partir d’une photo.

Tâches réalisées:

  • Préparation et augmentation des données ;
  • comparaison de différentes architectures ;
  • utilisation de transfert d’apprentissage à partir de plusieurs modèles pré-entraînés ;
  • développement d’une web app (disponible ici -> https://dogs-prediction.herokuapp.com/)

Configuration:

Environnement : Colaboratory notebook, GPU

Librairies : pandas, numpy, matplotlib, tensorflow, keras, tensorflowhub, os, tarfile, PIL