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
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.
- Clone the repo.
- Run:
pip install -r requirements.txt python3 setup.py
Remove the mislabeled and confusing images from the dataset to increase the accuracy of the classifying the scenes.