Updated
Browse files
README.md
CHANGED
|
@@ -1,3 +1,81 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Fashion MNIST Classifier Zoo 👕🎽👖
|
| 2 |
+
|
| 3 |
+
## Model Overview
|
| 4 |
+
|
| 5 |
+
Welcome to the Fashion MNIST Classifier Zoo! This model card showcases a collection of image classification models trained on the [Fashion MNIST dataset](https://github.com/zalandoresearch/fashion-mnist). Each model offers a unique approach to identifying articles of clothing from 28x28 grayscale images. Explore the models below to find the perfect fit for your fashion needs!
|
| 6 |
+
|
| 7 |
+
## Models
|
| 8 |
+
|
| 9 |
+
### CNN_Fashion_MNIST
|
| 10 |
+
- **Architecture:** A custom Convolutional Neural Network (CNN) designed for efficient feature extraction and classification.
|
| 11 |
+
- **Size:** 1.5 MB
|
| 12 |
+
- **Use Case:** Ideal for resource-constrained environments or applications requiring fast inference.
|
| 13 |
+
|
| 14 |
+
### VGG16_Fashion_MNIST
|
| 15 |
+
- **Architecture:** Implementation of the classic VGG16 architecture, leveraging its deep layers for robust feature learning.
|
| 16 |
+
- **Size:** 184 MB
|
| 17 |
+
- **Use Case:** Suitable for applications where high accuracy is paramount, even at the cost of increased computational complexity.
|
| 18 |
+
|
| 19 |
+
### Xception_Fashion_MNIST
|
| 20 |
+
- **Architecture:** Employs the Xception architecture, known for its efficient use of parameters and strong performance.
|
| 21 |
+
- **Size:** 279 MB
|
| 22 |
+
- **Use Case:** A good balance between accuracy and computational efficiency, making it suitable for a wide range of applications.
|
| 23 |
+
|
| 24 |
+
## Interactive Demo
|
| 25 |
+
|
| 26 |
+
Unfortunately, this model card is static, but imagine the possibilities! If this were interactive, you could:
|
| 27 |
+
1. **Upload your own fashion images** and see how each model classifies them.
|
| 28 |
+
2. **Compare the models' performance** on a held-out test set with interactive visualizations.
|
| 29 |
+
3. **Adjust confidence thresholds** to explore the trade-off between precision and recall.
|
| 30 |
+
|
| 31 |
+
## Intended Use
|
| 32 |
+
|
| 33 |
+
These models are intended for:
|
| 34 |
+
- Educational purposes: Learning about image classification and deep learning architectures.
|
| 35 |
+
- Benchmarking: Comparing the performance of different models on the Fashion MNIST dataset.
|
| 36 |
+
- Inspiration: Providing a starting point for building more sophisticated fashion recognition systems.
|
| 37 |
+
|
| 38 |
+
## How to Use
|
| 39 |
+
|
| 40 |
+
1. **Load the Model:** Use TensorFlow/Keras to load the `.keras` model file of your choice.
|
| 41 |
+
|
| 42 |
+
```
|
| 43 |
+
from tensorflow import keras
|
| 44 |
+
model = keras.models.load_model('VGG16_Fashion_MNIST.keras')
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
2. **Prepare Your Data:** Ensure your input data consists of 99x99x3 RGB images, preprocessed to match the model's expected input.
|
| 48 |
+
|
| 49 |
+
3. **Make Predictions:** Use the loaded model to predict the class of each image.
|
| 50 |
+
|
| 51 |
+
```
|
| 52 |
+
predictions = model.predict(your_test_data)
|
| 53 |
+
```
|
| 54 |
+
|
| 55 |
+
## Training
|
| 56 |
+
|
| 57 |
+
The `fashion_mnist.ipynb` notebook provides a complete guide to training these models from scratch. Follow the instructions in the notebook to:
|
| 58 |
+
1. **Load the Fashion MNIST dataset.**
|
| 59 |
+
2. **Preprocess the data.**
|
| 60 |
+
3. **Build and train the models.**
|
| 61 |
+
4. **Evaluate their performance.**
|
| 62 |
+
|
| 63 |
+
## Files
|
| 64 |
+
|
| 65 |
+
- `.gitattributes`: Specifies attributes for files in the repository.
|
| 66 |
+
- `CNN_Fashion_MNIST.keras`: Pre-trained CNN model.
|
| 67 |
+
- `VGG16_Fashion_MNIST.keras`: Pre-trained VGG16 model.
|
| 68 |
+
- `Xception_Fashion_MNIST.keras`: Pre-trained Xception model.
|
| 69 |
+
- `fashion_mnist.ipynb`: Jupyter Notebook for training and evaluation.
|
| 70 |
+
- `README.md`: This model card.
|
| 71 |
+
|
| 72 |
+
## Limitations and Future Directions
|
| 73 |
+
|
| 74 |
+
- **Dataset Bias:** The Fashion MNIST dataset is a simplified representation of real-world fashion images. Models trained on this dataset may not generalize well to more complex scenarios.
|
| 75 |
+
- **Limited Architectures:** This collection includes only a few popular architectures. Future work could explore more recent and advanced models.
|
| 76 |
+
- **No Interactive Demo:** As mentioned above, an interactive demo would greatly enhance the user experience.
|
| 77 |
+
|
| 78 |
+
## Author
|
| 79 |
+
|
| 80 |
+
[Harsh Maniya](https://huggingface.co/harshhmaniya)
|
| 81 |
+
[GitHUb](https://github.com/harshhmaniya)
|