Skip to content

puneet29/IntelSceneClassification

Repository files navigation

Intel Scene Classification Challenge

The problem statement is available on analytics vidhya. The problem is to classify scene depicted in the image. Try out the jupyter notebook here or the demo website: intelclassify.herokuapp.com. An example is given below

Example image

Download the pretrained model

Download the pretrained model from this google drive link and paste it in the repo folder. I have used RESNET-50 and DENSENET-161 architecture and added two fully-connected layers at the end of the architecture.

RESNET is used to clean out the dataset, while the DENSENET is used to train on the final cleansed dataset.

Setup

  • Clone the repo.
  • Run:
    pip install -r requirements.txt
    python3 setup.py

Key step

Remove the mislabeled and confusing images from the dataset to increase the accuracy of the classifying the scenes.