This project is an image classification web application that allows users to upload images and get predictions using a pre-trained neural network model.
The Dataset folder contains all the data required for training, validation, and testing:
- originalTrain – Dataset used to train the model.
- originalValidation – Dataset used to validate the model during training.
- originalTest – Dataset used to test the model’s performance via the Web App.
⚠️ It is recommended not to modify the Dataset folder, as it may break file path configurations.
Here’s a breakdown of the important files and their purposes:
- Training File – Performs image processing on the training dataset.
- Validation File – Performs image processing on the validation dataset.
- Neural Network File – Trains the model using processed training data and validates using processed validation data.
- Testing File – Processes images uploaded by users through the Web App and predicts results using the trained model.
- main.py – Main entry point connecting the Web App to the model. This is the file you should run.
- Model Folder – Contains the trained model, which is loaded during prediction.
- Dataset Folder – Contains all datasets (do not modify).
- template Folder – Contains HTML templates for the Web App.
- static Folder – Contains static files (CSS, JS, images) needed for the Web App.
Since the model is already trained, there is no need to run the training or validation scripts.
- Run the main file:
python main.py- Copy the link provided in the console and open it in your browser.
- Upload images and view predictions instantly!
- Muhammed Luqman
- Wafiya Sohail
- Farhan Shoukat