Skip to content

This is a desktop application built with Python and uses a Convolutional Neural Network (CNN) to classify images. The app is able to classify images into two categories, hamburgers or boxes. The main libraries used are Tkinter, Keras, and scikit-learn

Notifications You must be signed in to change notification settings

albertoecf/ml_gui_desktop

Repository files navigation

Image Classifier

This is a desktop application that utilizes a Convolutional Neural Network (CNN) to classify images into defined categories. In this example, we are categorizing images of hamburgers and boxes.

The main libraries used in this project are Tkinter (for building the GUI), Keras (for building and training the CNN), and Scikit-learn (for data preprocessing and model evaluation).

The CNN model was trained on a dataset of images of hamburgers and boxes, and achieved high accuracy in distinguishing between the two classes

Use case

This application can be used in various scenarios such as in online orders where the user orders a product and receives a different product than what was specified. This implementation may not seem very useful (who would be interested in distinguishing a box from a hamburger?). But it builds the architecture for more sophisticated implementations.

Features

This application allows users to:

  • Load an image from their computer system
  • View the image
  • Rate the image

Dependencies

The main libraries used in this project are:

  • Tkinter
  • Keras
  • scikit-learn

Running the application

To run the application, simply run the gui_image_classifier.py file. This will launch a GUI where users can upload their images and classify them

Example

Instantiate desktop app

Captura de pantalla 2022-12-31 a las 11 32 34

Upload local image :

Captura de pantalla 2022-12-31 a las 11 34 37

Run and Predict class :

Captura de pantalla 2022-12-31 a las 11 35 09

About

This is a desktop application built with Python and uses a Convolutional Neural Network (CNN) to classify images. The app is able to classify images into two categories, hamburgers or boxes. The main libraries used are Tkinter, Keras, and scikit-learn

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages