kolkela/simple-cnn-classifier

Simple CNN Classifier

Model Architecture

• Data Augmentation: Random Flip, Rotation, Zoom, Contrast
• Conv Blocks: 32 -> 64 -> 128 filters + MaxPool
• Regularization: Dropout
• Head: GlobalAveragePooling2D -> Dense Layer

How to Load & Use

import keras

model = keras.saving.load_model("hf://kolkela/simple-cnn-classifier")