Skip to content

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.

Notifications You must be signed in to change notification settings

NANDINISHARMA30/AlzheimerAugment-CNN-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

AlzAugment -THE CNN MODEL

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.

πŸ“ Project Structure

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

Model Layers:

Conv2D(128) β†’ BatchNorm β†’ MaxPool
Conv2D(64)  β†’ BatchNorm β†’ MaxPool
Conv2D(32)  β†’ BatchNorm β†’ MaxPool
β†’ Flatten β†’ Dense(128) β†’ Dropout(0.5) β†’ Dense(4, softmax)

Run the Model:

python model.py

About

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.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published