https://github.com/vinayakchandra/waste-wizard

Flask application that allows users to upload images of waste items and classify them into different categories.
https://github.com/vinayakchandra/waste-wizard

flask tensorflow

Last synced: about 2 months ago
JSON representation

Flask application that allows users to upload images of waste items and classify them into different categories.

• Host: GitHub
• URL: https://github.com/vinayakchandra/waste-wizard
• Owner: vinayakchandra
• Created: 2024-10-06T10:30:18.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-11-10T13:28:21.000Z (over 1 year ago)
• Last Synced: 2025-01-30T00:33:37.085Z (over 1 year ago)
• Topics: flask, tensorflow
• Language: Jupyter Notebook
• Homepage:
• Size: 53.5 MB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Waste-Wizard

## Description

`Flask` application that allows users to upload images of waste items and classify them into

categories such as batteries, clothes, e-waste, glass, light bulbs, organic. The app uses a

`TensorFlow` model for image classification and provides users with information and video resources related to the type

of waste.

## Features

- Upload images of waste items.

- Classify waste items using a pre-trained TensorFlow model.

- Display relevant information and video resources based on the classified category.

- User-friendly interface for uploading images and viewing results.

## Libraries Used 

- `Flask`: create the server, handle HTTP requests, manage routing

- `TensorFlow`: load a pre-trained model (`classifyWaste.h5`)

- `NumPy`: convert the uploaded images into arrays, normalize pixel values, and prepare them for input into the TensorFlow model for classification.

## Endpoints

| Endpoint               | Method | Description                                                                   |

|------------------------|--------|-------------------------------------------------------------------------------|

| `/`                    | GET    | Displays the main page for uploading images.                                  |

| `/`                    | POST   | Handles image uploads and classifies the uploaded image.                      |

| `/classify/` | GET    | Classifies the uploaded image and returns the predicted category and details. |

| `/display/`  | GET    | Displays the uploaded image.                                                  |

### Endpoint Details

* **`/` (GET)**: This is the home endpoint where the main upload interface is displayed. Users can upload an image here.

* **`/` (POST)**: This endpoint processes the uploaded image. It checks if a file is uploaded, validates the file type,

  saves the image, and then classifies it. After processing, it redirects to the home page and displays the

  classification results.

* **`/classify/` (GET)**: This endpoint is called to classify the uploaded image based on its path. It returns

  the predicted waste category and additional details related to that category.

* **`/display/` (GET)**: This endpoint redirects to the location of the uploaded image for display purposes.