Skip to content

Project performed as part of the exam of Artificial Neural Networks and Deep Learning at Politecnico di Milano. Goal of it was to create and train a Neural Network capable of distinguish between 14 different classes of images of leaves.

Notifications You must be signed in to change notification settings

GabrieleRivi/Leaves_Images_Classificator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

Leaves Images Classificator

Description of the problem

This small project was performed as an assignment for the course of Artificial Neural Networks & Deep Learning and it's part of the exam. The task was to develop a model for Image Classification and to train it to distinguis between 14 classes of leaves, as in the example below.

Schermata 2021-11-29 alle 14 37 38

Dataset Description

The dataset we used was composed of 17728 images, but training with only those resulted in a poor accuracy on the private test set. We used the ImageDataGenerator class to perform data augmentation.

Final Model

The ultimate model we submitted reached an accuracy of 92.08% on the private test set and was obtained using the technique of Transfer Learning; the model was then fine-tuned for two times, with a progressively smaller learning rate. Find more accurate description on the development and the specific details of the models on the attached document.

Group components:

Group Name: Gamma

About

Project performed as part of the exam of Artificial Neural Networks and Deep Learning at Politecnico di Milano. Goal of it was to create and train a Neural Network capable of distinguish between 14 different classes of images of leaves.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages