[go: nahoru, domu]

Skip to content

Homework 1 for the Deep Learning course of the MSc in Mathematical Engineering @ Polimi (A.Y. 2022-2023).

Notifications You must be signed in to change notification settings

teobucci/CNN-Plants-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional Neural Network: Plants Species Classification

This project was developed for the course of Artificial Neural Networks and Deep Learning for the MSc. in Mathematical Engineering at Politecnico di Milano, A.Y. 2022/2023.

Description

.
├── README.md
├── misc
│   ├── accuracy_results.csv
│   ├── confusion.pdf
│   ├── ensemble_vgg16-resnet50.png
│   ├── single_augmentation.jpg
│   ├── supernet.pdf
│   └── ypred.npy
├── models
├── notebooks
│   ├── final_model.ipynb
│   ├── keras_tuner.ipynb
│   ├── metrics.ipynb
│   ├── supernet_choice.ipynb
│   └── supernet_histories
├── report
│   ├── bibliography.bib
│   ├── report.pdf
│   └── report.tex
├── requirements.txt
└── training_data_final
  • final_model.ipynb is the main file that performs training and fine-tuning, and saves the models. Its decisions are partially based on the results from supernet_choice.ipynb and keras_tuner.ipynb. Finally, these models are evaluated in metrics.ipynb.
  • keras_tuner.ipynb provides the implementation of the Keras tuner code to tune the hyperparameters of the models.
  • metrics.ipynb contains the code to evaluate the models and provide accuracies and F1-scores.
  • supernet_choice.ipynb performs a pilot run of transfer learning testing different pretrained models.
  • accuracy_results.csv contains a table of the accuracies of some of the models that we trained during the competition, it's included for the sake of completeness and as appendix in case the report explanations wouldn't be clear enough.

Authors

Output

Check out the final report.pdf.

License

GNU GPLv3

About

Homework 1 for the Deep Learning course of the MSc in Mathematical Engineering @ Polimi (A.Y. 2022-2023).

Topics

Resources

Stars

Watchers

Forks