swaraj-khan/shape-recognition

This is a Python project that trains a deep learning model to identify three user-defined shapes based on an image drawn by the user. The application uses SKLEARN and a CV2 to predict the shape based on its features.

Shape-Recognition

This project allows the user to name three different shapes of their choosing and trains a machine learning model to identify those shapes based on an image drawn by the user.

The application uses SKLEARN and a CV2 to train a convolutional neural network (CNN) on a dataset of hand-drawn shapes. The trained model can then predict the user-drawn shape based on its features, such as the number of edges and angles.

The user interface is designed using the Tkinter library and provides an intuitive drawing canvas where the user can draw their shape using the mouse or touchpad. Once the drawing is complete, the user can submit the image to the model for prediction, and the application will display the name of the recognized shape.

This project is a great way to explore the world of deep learning and computer vision, and can be easily extended to recognize more shapes or even other types of images. So go ahead, give it a try and see if the model can recognize your shapes accurately!

Given below is the process flow :

Step 1: Enter the name of classifier

Step 2: Enter the names of 3 shapes

Step 4: Draw shapes with the brush. ( you can also increase or decrease the thickness of the brush + clear the canvas with the "clear" button if you are not happy ) and then press the button that has the name of the shape

Step 5: Once you are done with drawing shapes, press the "Train Model" button

Step 6: Now simply draw any shape and then press the "Predict" button to see which shape you have drawn