This is a Streamlit web application that classifies clothing items into various categories. Users can upload an image of clothing, and the app will use three different models to predict the category of the item.
The app utilizes three different models for clothing classification, each with its own architecture:
- Architecture:
- Convolutional Layer 1: 32 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Flatten Layer
- Dense Layer: 128 units, ReLU activation
- Output Layer: 10 units, Softmax activation
- Architecture:
- Convolutional Layer 1: 32 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Convolutional Layer 2: 64 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Flatten Layer
- Dense Layer 1: 128 units, ReLU activation
- Dropout Layer: 0.25
- Dense Layer 2: 256 units, ReLU activation
- Dropout Layer: 0.25
- Dense Layer 3: 128 units, ReLU activation
- Output Layer: 10 units, Softmax activation
- Architecture:
- Convolutional Layer 1: 64 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Convolutional Layer 2: 128 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Convolutional Layer 3: 64 filters, kernel size 3x3, ReLU activation
- MaxPooling Layer: 2x2
- Flatten Layer
- Dense Layer 1: 128 units, ReLU activation
- Dropout Layer: 0.25
- Dense Layer 2: 256 units, ReLU activation
- Dropout Layer: 0.5
- Dense Layer 3: 256 units, ReLU activation
- Dropout Layer: 0.25
- Dense Layer 4: 128 units, ReLU activation
- Dropout Layer: 0.10
- Output Layer: 10 units, Softmax activation
To run Predictstock locally, follow these steps:
-
Clone this repository to your local machine.
git clone https://github.com/rajsahu2004/Fashion-Item-Classifier.git
-
Install the requirements:
pip install -r requirements.txt
-
Run the Streamlit app.
streamlit run app.py
-
Open a web browser and access the app at
https://localhost :8501
-
Access the app in your web browser.
-
Upload an image of clothing.
Contributions are welcome! If you'd like to enhance Predictstock or fix any issues, please fork the repository and submit your pull requests.
This project is licensed under the MIT License - see the LICENSE file for details.