Animal Recognition Model

Model Overview

This model is designed to classify images of animals into predefined categories. It uses a ResNet50V2 base model and has been trained on a custom dataset.

Classes

The model was trained on the following classes:

cat
dog
horse
lion
tiger
elephant

Usage

Load the model using TensorFlow/Keras.
Preprocess the input image to a size of 256x256 and normalize it.
Pass the preprocessed image to the model for prediction.

from keras.models import load_model
import numpy as np
from tensorflow.keras.utils import load_img, img_to_array

def predict_image(image_path, model):
    img = load_img(image_path, target_size=(256, 256))
    img_array = img_to_array(img) / 255.0
    img_array = np.expand_dims(img_array, axis=0)
    prediction = model.predict(img_array)
    return np.argmax(prediction, axis=1)

model = load_model('best_model.weights.h5')
predicted_class = predict_image('image.jpg', model)
print(f"Predicted class: {predicted_class}")

Training Details

Base Model: ResNet50V2 (pretrained on ImageNet)
Dataset: Custom animal dataset
Optimizer: Adam
Loss Function: Sparse Categorical Crossentropy
Metrics: Accuracy
Augmentation: Applied during training

Model Performance

Will be updated soon

Downloads last month: -