Giter VIP home page Giter VIP logo

ajnavneet / apple-quality-segmentation-tensorflow-sagemaker-unet Goto Github PK

View Code? Open in Web Editor NEW
0.0 1.0 0.0 2.04 MB

Image segmentation model for checking apple quality using UNet model architecture on TensorFlow along with AWS SageMaker deployment.

License: MIT License

Python 2.00% Jupyter Notebook 98.00%
aws-s3 computer-vision fruit-recognition sagemaker-deployment unet-image-segmentation

apple-quality-segmentation-tensorflow-sagemaker-unet's Introduction

Fruit Quality using UNet Image Segmentation and AWS SageMaker deployment

Overview

Amazon SageMaker is a machine learning service that empowers data scientists to train and deploy machine learning models efficiently. It offers an integrated Jupyter authoring notebook instance for data exploration and analysis, as well as streamlined model deployment in a hosted environment, reducing reliance on software engineers.

SageMaker supports data preprocessing, post-processing, feature engineering, and model evaluation.


Aim

In this project, we will build an image segmentation model using TensorFlow on Amazon SageMaker, utilizing the UNet model architecture.


Tech Stack

  • Language: Python
  • Libraries: TensorFlow, Pandas, NumPy, Scikit-learn, Patchify, Matplotlib, Segmentation Models, Boto3

Data Description

The dataset contains images of rotten apples with annotated areas of rot. These images will be used to train the model.


Approach

  1. Data Loading

  2. Data Preprocessing

    • Image Patching
  3. Model Building and Training

    • Segmentation model in TensorFlow
  4. Model Deployment on Amazon SageMaker


Modular Code Overview

  1. Input: Contains the data for analysis, including images of rotten apples.
  2. Notebook: Contains the Jupyter notebook file for the project.
  3. MLPipeline: A folder with functions placed in different Python files, appropriately named. These functions are called inside the "Engine.py" file.
  4. Sagemaker_Deployment: Contains files related to deployment.
  5. requirements.txt: Lists all required libraries with respective versions. Install them using the command pip install -r requirements.txt.
  6. Readme.md: Provides instructions for running the code.

apple-quality-segmentation-tensorflow-sagemaker-unet's People

Contributors

ajnavneet 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.