Skip to content

Tomato Status Classification using Deep learning with different preprocessing strategies

License

Notifications You must be signed in to change notification settings

KohWenLe/Tomato-Status-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🍅Tomato Status Classification using Deep Learning with Different Image Preprocessing Techniques

This repository contains experiments for tomato status (ripeness) classification using deep learning and different image preprocessing techniques. The goal is to evaluate how preprocessing methods affect model performance when classifying tomatoes into ripe, unripe, and rotten categories. This experiment is conducted for academic purpose.


Models Used

  • ResNet50 (transfer learning)
  • VGG16 (transfer learning)
  • Custom CNN

Image Preprocessing Methods

The following preprocessing strategies are evaluated:

  • No preprocessing (original images)

1

  • Morphological processing

2

  • Grayscale conversion

  • Gaussian blur

  • Thresholding

  • Morphological opening and closing

  • CLAHE + K-Means segmentation

3

  • LAB color space conversion
  • K-Means clustering (k = 3)
  • CLAHE for contrast enhancement

Files in This Repository

model training and evaluation/
├── CNN CLAHE + kmeans.ipynb
├── Resnet50 original.ipynb
├── Resnet50 morphology.ipynb
├── Resnet50 CLAHE + kmeans.ipynb
└── VGG16 CLAHE + kmeans.ipynb

Key Results

  • Best preprocessing method: CLAHE + K-Means segmentation
  • Best performing model: VGG16
  • Highest test accuracy: 95.09%

Dataset

  • Tomato images from a public dataset and self-collected data
  • Classes: ripe, unripe, rotten
  • Dataset is not included in this repository

Purpose

This project was developed for an academic Computer Vision course to study the impact of image preprocessing on deep learning–based classification


About

Tomato Status Classification using Deep learning with different preprocessing strategies

Resources

License

Stars

Watchers

Forks

Packages

No packages published