kolkela's picture
Update README.md
27ac963 verified
|
Raw
History Blame Contribute Delete
454 Bytes
metadata
language: en
tags:
  - keras
  - computer-vision
  - image-classification
  - cnn
license: mit

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")