This project is a simple clothing classifier web application built using Streamlit and TensorFlow. It leverages a custom-trained Fashion MNIST model to classify images of clothing items. The application provides real-time predictions along with a bar chart visualizing the confidence levels for each class.
- Upload images of clothing items in JPG, JPEG, or PNG format.
- Classify the uploaded image using a custom-trained Fashion MNIST model.
- Display the predicted clothing category.
- Visualize the confidence levels of all possible classes in a bar chart.
The Fashion MNIST dataset contains 60,000 training images and 10,000 testing images, each belonging to one of the 10 clothing categories. These are:
- T-shirt/top
- Trouser
- Pullover
- Dress
- Coat
- Sandal
- Shirt
- Sneaker
- Bag
- Ankle boot
These categories represent the different types of clothing that the model has been trained to recognize.
- Python 3.x
- TensorFlow
- Streamlit
- Pandas
- PIL (Pillow)
- Numpy
-
Navigate to the project directory.
cd fashion_mnist-classifier/app
-
Install dependencies:
pip install streamlit
-
Start the Streamlit app:
streamlit run main.py
-
Open the application in your web browser at
http://localhost:8501
.
- Upload an image of a clothing item using the file uploader.
- Click the "Classify" button to get the predicted category.
- View the prediction along with a bar chart showing the confidence levels for each class.
main.py
: The main Streamlit application file.trained_model/
: Directory containing the custom-trained model file (trained_fashion_mnist_model.h5
).