https://github.com/its-kumar/indian_celebrity_image_classifier

https://github.com/its-kumar/indian_celebrity_image_classifier

Last synced: about 1 month ago
JSON representation

• Host: GitHub
• URL: https://github.com/its-kumar/indian_celebrity_image_classifier
• Owner: its-Kumar
• Created: 2020-09-13T04:45:25.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2020-09-13T05:35:14.000Z (over 5 years ago)
• Last Synced: 2026-01-01T02:35:31.676Z (4 months ago)
• Language: Jupyter Notebook
• Size: 22.2 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.MD

Awesome Lists containing this project

README

          
# Indian Celebrity Image Classifier

It is a Machine Learning Project on Image Classification Problem. I have built this project using `python flask server` for back-end and `HTML, CSS, JS` for front-end UI.

Basically it is a drag and drop UI. you can drag any pic and click on `clasify` button and the model will predict which celebrity's face the pic has. This model will predict the result for only these Indian celebrities.

1. Akshay Kumar

2. Honey Singh

3. Sachin Tendulkar

4. Sunny Leone

5. Virat Kohli

## UI

![pic](app.png)

---

## Model

Used `opencv` for **Face Detection**.

I have used some classification models. below are the results of score of the models.

  

    

      

      model

      best_score

      best_params

    

  

  

    

      0

      svm

      0.741667

      {'svc__C': 1, 'svc__kernel': 'linear'}

    

    

      1

      random_forest

      0.606159

      {'randomforestclassifier__n_estimators': 10}

    

    

      2

      logistic_regression

      0.734058

      {'logisticregression__C': 10}

    

  

## Created by KUMAR SHANU