{{ message }}
puneet29/IntelSceneClassification
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Repository files navigation
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.
