This project aims to build an image classification model using PyTorch that can accurately classify images into one of six predefined categories. This project will utilize the Intel Image Classification Dataset, which contains a diverse set of images suitable for training and testing classification.
Data Source The dataset for this project is the Intel Image Classification Dataset, which can be found on Kaggle. It consists of natural scene images that are organized into six different categories. The dataset can be accessed and downloaded from the following link: https://www.kaggle.com/puneet6060/intel-image-classification
NOTE: Only a subset of the dataset was used for this training, remember to increase the amount of data as you like. Point the training and test folder path to the dataset path on your end