Skip to content

Faisal-javed-khan/classification-using-Machine-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

Here is the detailed HTML version of the text:

Project Title

Table of Contents

Introduction

This project aims to develop an image classification system using machine learning techniques. The goal is to classify images into three categories: Car, Cone, and Ball. The objectives are to:

  • Explore feature extraction techniques (SIFT, ORB, HOG)
  • Train machine learning models (Logistic Regression, Decision Tree, Random Forest, SVM)
  • Evaluate model performance

Background

Image classification is a fundamental task in computer vision. It involves assigning a label to an image based on its content. Machine learning techniques have shown great promise in image classification tasks.

Problem Statement

The problem is to develop an image classification system that can accurately classify images into three categories: Car, Cone, and Ball.

Methodology

We collected a dataset of images and extracted features using SIFT, ORB, and HOG. We then trained four machine learning models and evaluated their performance using accuracy metrics.

Data Collection

We collected a dataset of 1000 images, with 500 images per class.

Feature Extraction

We extracted features from the images using SIFT, ORB, and HOG.

Model Training

We trained four machine learning models: Logistic Regression, Decision Tree, Random Forest

Here is the rest of the HTML version of the text:

Model Evaluation

We evaluated the performance of each model using accuracy metrics.

Results

The results show that HOG features with Logistic Regression achieved the highest accuracy of 90%.

Accuracy Comparison

The accuracy of each model is shown in the table below:

Model Accuracy
Logistic Regression 90%
Decision Tree 85%
Random Forest 88%
SVM 82%

Discussion

The results indicate that HOG features are effective in image classification. Logistic Regression outperformed other models, likely due to its simplicity and ability to handle high-dimensional data.

Conclusion

This project demonstrated the effectiveness of HOG features and Logistic Regression in image classification. Future work should explore other feature extractors and deep learning models.

Recommendations

  • Explore other feature extractors (e.g., CNN-based features)
  • Investigate deep learning models (e.g., CNN, ResNet)
  • Evaluate performance on a larger dataset

About

Machine learning handcrafted Features extraction and Classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors