harshhmaniya commited on
Commit
6d616a5
·
verified ·
1 Parent(s): b41d3e5
Files changed (1) hide show
  1. README.md +81 -3
README.md CHANGED
@@ -1,3 +1,81 @@
1
- ---
2
- license: apache-2.0
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)