I initiated the project of building the image classifier that classifies images of forests from cities.
This project was all about building image classification model that identifies whether the given image is the forest or city.
As a part of the dataset, I took both forest and street views from the cities, like
1. Pune, India
2. Accra, Ghana
3. Stirling, United Kingdom
I tried our datasets on lots of pre-trained computer vision models, for instance, VGGNet, AlexNet, and ResNet. Surprisingly, these models gave very low accuracy for our datasets.
Post that, we tried Yolo v5 model on our datasets. The model performed well on my datasets.
Further, I built my own customized convolutional neural network and tried the same dataset on that, which turned out fantastic. The performance parameters of this model were satisfactory.
For reports and readings, kindly visit the link:
https://github.com/surajjeoor/ITNPAI_Project_Forestcityclassifier
0 comments:
Post a Comment