Animal Classification using Transfer Learning Model (MobileNetV2)

Image Dataset Link: https://www.kaggle.com/datasets/alessiocorrado99/animals10

Overview

This is a image classification web application, using Streamlit. This app will classify 6 animals(dog, horse, elephant, butterfly, chicken, cat).

Procedure

• Install given modules
• Build Model
• Build Model
• Test Results

Model

I have chosen the pretrained "MobileNetV2" model to perform classification.

Modules Required

• streamlit==1.14.0
• numpy==1.23.4
• opencv-python==4.6.0.66
• tensorflow==2.10.0

Streamlit Web App

After uploading an image, this is how result show.