Skip to content

๐Ÿพ This project builds a deep learning model to classify animals from images using transfer learning and CNNs. It processes visual data, predicts species with high accuracy, and presents results through an interactive dashboardโ€”ideal for ecological research, education, and real-time applications.

License

Notifications You must be signed in to change notification settings

SBanditaDas/Real-Time-Image-Classification-of-Animals

Repository files navigation


๐Ÿงพ Animal Classification โ€“ Image-Based Deep Learning Model


Identifying animal species from images using Convolutional Neural Networks (CNNs), data augmentation, and transfer learning for robust classification.


๐Ÿ“Œ Table of Contents


Overview

This project classifies animals based on image inputs using a deep learning pipeline built with TensorFlow and Keras. It leverages transfer learning with pre-trained models to achieve high accuracy across multiple species, and includes visualizations for performance metrics and prediction confidence


Project Problem

Accurate animal classification is essential for wildlife monitoring, conservation, and educational tools. This project aims to:

  • Automate species identification from images
  • Reduce manual effort in ecological surveys
  • Improve classification accuracy across diverse animal categories
  • Enable real-time prediction for deployment in mobile/web apps

Dataset

  • Image dataset stored in /data/ folder with subfolders per class (e.g., lion, elephant, zebra)
  • Includes training, validation, and test sets
  • Augmented using rotation, zoom, flip, and brightness adjustments

Tools & Technologies

  • Python (TensorFlow, Keras, OpenCV, Matplotlib)
  • Google Colab (GPU acceleration)
  • GitHub
  • Streamlit (for deployment)

Project Structure

animal_classification/
โ”‚
โ”œโ”€โ”€ README.md
โ”œโ”€โ”€ .gitignore
โ”œโ”€โ”€ requirements.txt
โ”œโ”€โ”€ Classification of animals.pdf
โ”‚
โ”œโ”€โ”€ notebooks/                                # Jupyter notebooks
โ”‚   โ”œโ”€โ”€ exploratory_data_analysis.ipynb
โ”‚   โ”œโ”€โ”€ animal-img-clsf.ipynb
โ”‚
โ”œโ”€โ”€ summery/                  
โ”‚   โ”œโ”€โ”€ animal_classifier_model.pkl
โ”‚   โ””โ”€โ”€classification_report.txt
โ”‚
โ”œโ”€โ”€ Outputs/                  # visuals file
โ”‚   โ””โ”€โ”€ airborne.png
โ”‚    โ””โ”€โ”€ aquatic.png

Data Cleaning & Preparation

  • Removed corrupted or mislabeled images
  • Resized all images to 224x224 pixels
  • Applied data augmentation to balance classes
  • Encoded labels and split into train/val/test sets

Exploratory Data Analysis (EDA)

Class Distribution:

  • 10 animal classes with varying image counts
  • Balanced using augmentation for minority classes

Image Quality Checks:

  • Verified resolution consistency
  • Removed grayscale and low-contrast images

Sample Visuals:

  • Grid plots of sample images per class
  • Distribution bar chart of image counts

Research Questions & Key Findings

  • Model Accuracy: Achieved 94.2% accuracy on test set using ResNet50
  • Misclassifications: Most confusion between tiger vs. leopard due to similar patterns
  • Augmentation Impact: Improved accuracy by 7.5% on minority classes
  • Confidence Scores: Average prediction confidence = 0.91
  • Real-Time Prediction: Streamlit app delivers <1s prediction latency

Graphs/Plots

Aquatic_Animal graphs Airborne_Animal graphs

How to Run This Project

  1. Clone the repository:
git clone https://github.com/SBanditaDas/animal-classification.git
  1. Open and run notebooks:

    • animal-img-clsf.ipynb
  2. View results:

Check - predictions.csv - classification_report.txt


Author & Contact

Sushree Bandita Das

S_Bandita_Das sushree-bandita-das-160651309 SBanditaDas dasbanditasushree

About

๐Ÿพ This project builds a deep learning model to classify animals from images using transfer learning and CNNs. It processes visual data, predicts species with high accuracy, and presents results through an interactive dashboardโ€”ideal for ecological research, education, and real-time applications.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors