Instructions to use kolkela/simple-cnn-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use kolkela/simple-cnn-classifier with Keras:
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://kolkela/simple-cnn-classifier") - Notebooks
- Google Colab
- Kaggle
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")
- Downloads last month
- 12