This project is focused on creating a Convolutional Neural Network model to predict the animal from an picture.
- Build a Convolutional Neural Network (CNN) to classify images from a given dataset into predefined categories.
- Implement a transfer learning approach using a pre-trained model.
- Compare the performance of the custom CNN and the transfer learning model based on evaluation metrics and analysis.
All the project is documented the image_classificator_animals.ipynb file.
A review of of the project at the presentation CNNImageClasifier.pdf including the answer to the initial questions, implementationd details, obstacles and conclusions.
- Animals-10 by Corrado Alessio
The dataset is not included on the repository but the notebook contains a script to download it and set it up on the proper directory.
Animals-10 is under license GPL2
This work is licensed under a Creative Commons Attribution 4.0 International License.
