suvmkr's picture
Update README.md
5d4834d verified
metadata
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

from tensorflow.keras.models import load_model

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