https://github.com/davityak03/animal-image-classification-using-cnn
Animal image classification using Convolutional Neural Networks (CNNs) involves training a model to recognize and categorize different animal species based on their images. The CNN architecture excels at automatically extracting features from images, such as shapes and textures, which are crucial for accurate classification.
https://github.com/davityak03/animal-image-classification-using-cnn
cnn-classification keras numpy pandas pretrained-models python tensorflow
Last synced: about 2 months ago
JSON representation
Animal image classification using Convolutional Neural Networks (CNNs) involves training a model to recognize and categorize different animal species based on their images. The CNN architecture excels at automatically extracting features from images, such as shapes and textures, which are crucial for accurate classification.
- Host: GitHub
- URL: https://github.com/davityak03/animal-image-classification-using-cnn
- Owner: Davityak03
- Created: 2024-06-21T08:03:37.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-06-21T08:14:10.000Z (over 1 year ago)
- Last Synced: 2025-01-18T12:21:21.369Z (about 1 year ago)
- Topics: cnn-classification, keras, numpy, pandas, pretrained-models, python, tensorflow
- Language: Jupyter Notebook
- Homepage:
- Size: 3.47 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Animal Species Classification Using CNN
## Overview
Animage is a machine learning project that focuses on image classification using Convolutional Neural Networks (CNNs). The project aims to classify images into predefined categories with high accuracy by leveraging the powerful feature extraction capabilities of CNNs.
## Dataset
Link for dataset https://drive.google.com/drive/folders/1Oe7fsq0Ej3FUZqvl6lLmA4726Qq8E1qa?usp=sharing
## Features
- **Image Preprocessing**: Resizing, normalization, and data augmentation techniques to enhance the model's performance.
- **CNN Architecture**: A robust CNN model designed for image classification.
- **Training and Validation**: Efficient training pipeline with validation steps to monitor performance.
- **Performance Metrics**: Accuracy, precision, recall, and F1-score for evaluating the model.
## Requirements
- Python 3.7 or higher
- TensorFlow 2.0 or higher
- Keras
- NumPy
- Pandas
- Matplotlib
- scikit-learn
## Model Architecture
The CNN architecture consists of:
- **Convolutional Layers**: For feature extraction.
- **Pooling Layers**: For down-sampling.
- **Fully Connected Layers**: For classification.
The model is designed to handle the complexity of image data and achieve high classification accuracy.
## Data Augmentation
Data augmentation techniques such as rotation, zoom, and horizontal flipping are applied to enhance the model's generalization capability.
## Evaluation Metrics
- **Accuracy**: Measures the proportion of correctly classified images.
- **Precision**: Measures the accuracy of positive predictions.
- **Recall**: Measures the ability of the model to identify all relevant instances.
- **F1-Score**: Harmonic mean of precision and recall.