Git Product home page Git Product logo

dl_project's Introduction

Rice Image Classification Project

This project focuses on classifying different varieties of rice using machine learning techniques. The dataset used in this project contains images of five different types of rice: Arborio, Basmati, Ipsala, Jasmine, and Karacadag.

Dataset

The dataset used in this project is available on Kaggle: Rice Image Dataset. It consists of images of rice grains belonging to each variety.

Data Preparation

  • The dataset is organized into folders, with each folder representing a rice variety.
  • Images are preprocessed and resized to 224x224 pixels to fit the input shape required for the models.

Models Implemented

1. Custom Convolutional Neural Network (CNN)

  • A custom CNN model is built using Keras with the following architecture:
    • Input layer: Convolutional layer with ReLU activation
    • MaxPooling layer
    • Flatten layer
    • Dense layer with ReLU activation
    • Dropout layer
    • Output layer with softmax activation
  • The model is trained and evaluated on the dataset.

2. Basic Neural Network

  • A basic neural network model is implemented using Keras with the following architecture:
    • Flatten layer
    • Dense layer with ReLU activation
    • Output layer with softmax activation
  • The model is trained using SGD optimizer and evaluated on the dataset.

3. Transfer Learning with InceptionV3

  • Transfer learning is employed using the InceptionV3 pre-trained model.
  • The InceptionV3 model is loaded with pre-trained ImageNet weights and appended with additional layers.
  • The model is fine-tuned on the rice image dataset.

4. Transfer Learning with ResNet50

  • Transfer learning is also applied using the ResNet50 pre-trained model.
  • The base ResNet50 model is loaded with pre-trained ImageNet weights and extended with custom layers.
  • The model is trained on the rice dataset while keeping the base ResNet50 layers frozen.

Dependencies

  • Python 3.x
  • TensorFlow 2.x
  • Keras
  • Matplotlib
  • Seaborn
  • NumPy
  • Pandas

Usage

  1. Download the dataset from Kaggle and place it in the project directory.
  2. Run the provided Jupyter notebook or Python script to train and evaluate the models.

Results

  • The performance of each model is evaluated based on accuracy and loss metrics.
  • Model training progress is visualized using matplotlib.

Conclusion

This project demonstrates the application of various machine learning techniques for rice image classification. The models implemented achieve satisfactory accuracy in classifying different rice varieties.

dl_project's People

Contributors

pranav-69 avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.