/Pistachio-Classifier-Pytorch

Pistachio classifier model using Pytorch

Primary LanguageJupyter Notebook

Overview

The model uses both PyTorch and TensorFlow frameworks for training and evaluation. It aims to assist in the automatic recognition of different types of pistachios from images.

Some of the Predicitons : image

Features

  • Train and evaluate the model using both PyTorch and TensorFlow.
  • Accurately classify pistachios into Siirt and Kirmizi categories.
  • Handling and preprocessing of image datasets.

Dataset

The dataset was taken from Kaggle website " https://www.kaggle.com/datasets/muratkokludataset/pistachio-image-dataset " , The dataset consists of:

  • 700 images of Siirt pistachios
  • 1160 images of Kirmizi pistachios

Lisence

This project is licensed under the MIT License.