/Image-classifiaction-using-SVM

Celebrity Image Classification

Primary LanguageJupyter NotebookMIT LicenseMIT

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

  1. Maria Sharapova
  2. Serena Williams
  3. Virat Kohli
  4. Roger Federer
  5. Lionel Messi

Here is the folder structure,

  • UI : This contains ui website code
  • server: Python flask server
  • model: Contains python notebook for model building
  • google_image_scrapping: code to scrap google for images
  • images_dataset: Dataset used for our model training

Technologies used in this project,

  1. Python
  2. Numpy and OpenCV for data cleaning
  3. Matplotlib & Seaborn for data visualization
  4. Sklearn for model building
  5. Jupyter notebook, visual studio code and pycharm as IDE
  6. Python flask for http server
  7. HTML/CSS/Javascript for UI

Here is the video playlist for entire project: https://www.youtube.com/playlist?list=PLeo1K3hjS3uvaRHZLl-jLovIjBP14QTXc