fastai framework is used to classify natural scenes images, provided in the dataset. Particularly, transfer learning on resnet-34 and resnet-50 models (trained on Image-Net) is used.
This is image data of Natural Scenes around the world.
This Data contains around 25k images of size 150x150 distributed under 6 categories. {'buildings' -> 0, 'forest' -> 1, 'glacier' -> 2, 'mountain' -> 3, 'sea' -> 4, 'street' -> 5 }
The Train, Test and Prediction data is separated in each zip files. There are around 14k images in Train, 3k in Test and 7k in Prediction. This data was initially published on https://datahack.analyticsvidhya.com by Intel to host a Image classification Challenge.
Thanks to https://datahack.analyticsvidhya.com for the challenge and Intel for the Data.
Unfreeze model is trained with a variable learning rate ranging from 1e-6 to 1e-4.
It shows the model is most confused with buildings and streets, which is reasonable to some extent as the street view can have buildings also.
Unfreeze model is trained with a variable learning rate ranging from 1e-6 to 1e-4.
It also showed the same confusion as the previous model.
The notebook can be easily viewed here.