File size: 1,332 Bytes
fd22aaf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# Convolutional Neural Network (CNN) Model
This repository contains the configuration and weights for a Convolutional Neural Network (CNN) model trained on image data. The model architecture is defined using the Keras Sequential API.
## Model Architecture
The model is defined as a Sequential model with the following layers:
1. Input Layer
- Input shape: (None, 32, 32, 1)
2. Convolutional Layer
- Filters: 32
- Kernel size: (3, 3)
- Activation function: ReLU
- Batch normalization
- Max pooling: pool size (2, 2), strides (2, 2)
3. Dropout Layer
- Dropout rate: 0.25
4. Convolutional Layer
- Filters: 64
- Kernel size: (3, 3)
- Activation function: ReLU
- Batch normalization
- Max pooling: pool size (2, 2), strides (2, 2)
5. Dropout Layer
- Dropout rate: 0.25
- Flatten Layer
6. Dense Layer
- Units: 128
- Activation function: ReLU
- Batch normalization
7. Dropout Layer
- Dropout rate: 0.5
8. Dense Layer
- Units: 6 (output layer)
- Activation function: Softmax
## Categories to Predict
The model predicts images into the following categories:
- Accessories
- Bags
- Clothes
- Shoes
- Watches
## Model Files
- `model_config.json`: Configuration file containing the model architecture.
- `model_weights.h5`: File containing the model weights.
Feel free to use this model for your category classification tasks! |