Skip to content

Photo Classification and Image Tag Generation on Yelp dataset of 200k images

Notifications You must be signed in to change notification settings

chaitanyakasaraneni/yelpphotoclassification

Repository files navigation

Yelp Photo Classification

This project is done as a part CMPE-257 Machine Learning course.

Dataset used

The dataset used in this project is taken from the yelp website. It can be downloaded using the link: https://www.yelp.com/dataset

Models Used

Image Classification

2-layer CNN
Inception V3 model (with pretrained ImageNet weights)
SqueezeNet

Image Tagging

XGBoost
Tensorflow based 3-layer CNN