Skip to content

Commit

Permalink
Merge pull request #708 from Varunshiyam/Fixes-#704-Coffee
Browse files Browse the repository at this point in the history
Fixes #704 coffee Bean Classification
  • Loading branch information
UppuluriKalyani authored Nov 2, 2024
2 parents 7e832ac + cf92047 commit 1fea582
Show file tree
Hide file tree
Showing 2 changed files with 961 additions and 0 deletions.
44 changes: 44 additions & 0 deletions Prediction Models/Coffee_bean_Classification/Readme.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
# Coffee Bean Classification with Deep Learning

This project leverages deep learning to classify coffee bean types based on visual features. By training a convolutional neural network (CNN) model on labeled coffee bean images, we aim to create a system capable of identifying different types of coffee beans automatically.

## Table of Contents

- [Introduction](#introduction)
- [Project Structure](#project-structure)
- [Dataset](#dataset)
- [Model](#model)
- [Results](#results)

## Introduction

This project focuses on image classification using deep learning techniques to differentiate between various coffee bean types. This is particularly useful in industries that require high-quality control and consistent coffee bean quality.

## Project Structure

```plaintext
├── data/ # Dataset folder containing images
├── notebooks/ # Jupyter notebooks for experimentation
├── src/ # Source files for model training and evaluation
├── models/ # Saved models and checkpoints
├── results/ # Evaluation results and metrics
└── README.md # Project documentation
```

## Dataset
The dataset contains images of coffee beans from different types. These images are preprocessed for uniformity, and are then split into training, validation, and testing sets.

## Model
We implemented a convolutional neural network (CNN) model for image classification. Key layers include:

- Convolutional layers for feature extraction
- Pooling layers to reduce dimensionality
- Dense layers to classify images based on extracted features

## Results
The model's performance is evaluated using metrics such as accuracy, precision, recall, and F1-score on the test dataset. The model achieves satisfactory results in distinguishing between coffee bean types.





Large diffs are not rendered by default.

0 comments on commit 1fea582

Please sign in to comment.