image_classification_cnn

This repo contains the code of the project for the 2nd module of the course "Image Processing and Computer Vision" of the Master in AI at the University of Bologna. The project consists in classifying images of products in the Grocery Store Dataset using Convolutional Neural Networks.

Description

The project is divided into 2 parts:

• Build a CNN from scratch.
• Use a pre-trained model (ResNet18) and fine-tune it.

Authors

Christian Di Buò $\cdot$ Simone Reale