README.md

---
tags:
  - image-classification
  - cnn
  - keras
  - tensorflow
license: apache-2.0
datasets:
  - custom-dataset
model_creator: suvmkr
model_name: Classification Between 6 Types
framework: tensorflow
---

# Classification Between 6 Types

This is a Convolutional Neural Network (CNN) model trained to classify images into six categories.

## Model Details
- Framework: TensorFlow/Keras
- Model Format: `.keras`
- Input Image Size: `150x150`
- Classes: [buildings,forest,glacier,mountain,sea,street]

### Load the Model
```python
from tensorflow.keras.models import load_model

model = load_model("https://huggingface.co/suvmkr/classification_between_6_types/resolve/main/yes.keras")