CNN Image Classifier (4 Classes - Grayscale)
This repository contains a Convolutional Neural Network (CNN) implemented with TensorFlow/Keras to classify grayscale images into four classes (0, 1, 2, 3). It includes custom preprocessing, model training, and evaluation on unseen image data.
The CNN uses:
- 3 Conv2D layers with L2 regularization
- Batch Normalization and MaxPooling2D after each conv layer
- Flatten and Dense layers
- Dropout (0.5) to reduce overfitting
- Final output layer with softmax for 4-class classification
##DEPENDENCIES:
- tensorflow
- keras
- numpy
- pillow
- matplotlib
Conv2D(128) β BatchNorm β MaxPool
Conv2D(64) β BatchNorm β MaxPool
Conv2D(32) β BatchNorm β MaxPool
β Flatten β Dense(128) β Dropout(0.5) β Dense(4, softmax)python model.py