Celebrity-Image-Recognition

In this data science and machine learning project, we classify sports personalities. We restrict classification to only 5 people,

. Maria Sharapova

. Serena Williams

. Virat Kohli

. Rohit Sharma

. Mahendra Singh Dhoni

Here is the folder structure

1. UI : This contains ui website code

2. model: Contains python notebook for model building

3. model_dataset: Dataset used for our model training

4. server: Python flask server

5. google_images_webscraper: code to scrap google for images

Technologies used in this project,

. Python

. Numpy and OpenCV for data cleaning

. Matplotlib & Seaborn for data visualization

. Sklearn for model building

. Optuna for hyperparameter tuining and optimization

. Jupyter notebook, visual studio code and pycharm as IDE

. Python flask for http server

. HTML/CSS/Javascript for UI

The model used is an ensemble model including XGBoost,RandomForest,LGBM and Logistic Regression , trained on over 900 images!