DEV Community

Kavish Sanghvi
Kavish Sanghvi

Posted on • Updated on

Fauna Image Classification using Convolutional Neural Network

Aim

Aim of the project was to develop an animal image classifier in dense forest environments to achieve the desired accuracy, and aid ecologists and researchers in neural network, artificial intelligence, and zoological domains to further study and/or improve habitat, environmental and extinction patterns.

How I built it?

We present a methodology for the classification of fauna images, which will help ecologists and scientists to further study and/or improve habitat, environmental, and extinction patterns. We have used a Convolutional Neural Network with Leaky ReLU activation function and VGG16 architecture for our model. The initial step taken by the system aims at the creation of features with VGG16 model. Application of Image Processing along with Loading, Testing, Training, and Validating the dataset before the training step helps to remove the noise, obstacles, distortion and dirt from the images. The next step uses Convolutional Neural Network along with Leaky ReLU to train the model to accurately and precisely classify animal classes. In order to avoid the problem of Dying ReLU, where some ReLU neurons essentially die for all inputs and remain inactive no matter what input is supplied, here no gradient flows and if a large number of dead neurons are there in a neural network its performance is affected. To resolve this issue, we make use of what is called Leaky ReLU, where slope is changed left of x=0 and thus causing a leak and extending the range of ReLU. After training the model, we graph the model’s training and validation accuracy and loss to have insights about how well the model is trained. Lesser the loss, more is the accuracy. The next step is to generate classification matrix and confusion matrix to have exact details about how correctly the model is trained and classifying, as we cannot only rely on the accuracy. Lastly, we tested our model with sample data and found it to be accurately classified.

Dataset

Animal-10 dataset, which contains around 26179 hand-picked images of animals such as Butterfly, Cat, Chicken, Cow, Dog, Elephant, Horse, Sheep, Spyder, and Squirrel. Image count for each category varies from 2000 to 5000 images. Dataset was made available open-source through Kaggle.

Research Publication

The research paper "Fauna Image Classification using Convolutional Neural Network" is published by Science and Engineering Research Support soCiety in the International Journal of Future Generation Communication and Networking, indexed by Web of Science, J-Gate, Directory of Open Access Journals, and ProQuest.

Link to Code

GitHub logo kavishsanghvi / fauna-image-classification-using-convolutional-neural-network

Convolutional neural network for classification of animal images on Animal-10 dataset

Fauna Image Classification using Convolutional Neural Network

Convolutional neural network for classification of animal images from Animal-10 dataset

Aim

Aim of the project was to develop an animal image classifier in dense forest environments to achieve desired accuracy, and aid ecologists and researchers in neural network/Artificial Intelligence & zoological domains to further study and/or improve habitat, environmental and extinction patterns.

Method

We present a methodology for the classification of fauna images, which will help ecologist and scientists to further study and/or improve habitat, environmental and extinction patterns. We have used Convolutional Neural Network with Leaky ReLU activation function and VGG16 architecture for our model. The initial step taken by the system aims at creation of features with VGG16 model. Application of Image Processing along with Loading, Testing, Training, and Validating the dataset before the training step helps to remove the noise, obstacles, distortion and dirt from the images. The next…

Contact

Email | LinkedIn | Website | Medium | Twitter | Facebook | Instagram

Top comments (0)