Murdoch University Research Repository

Welcome to the Murdoch University Research Repository

The Murdoch University Research Repository is an open access digital collection of research
created by Murdoch University staff, researchers and postgraduate students.

Learn more

Deep learning approach to classify Tiger beetles of Sri Lanka

Abeywardhana, D.L., Dangalle, C.D., Nugaliyadde, A. and Mallawarachchi, Y. (2021) Deep learning approach to classify Tiger beetles of Sri Lanka. Ecological Informatics, 62 . Art. 101286.

Link to Published Version: https://doi.org/10.1016/j.ecoinf.2021.101286
*Subscription may be required

Abstract

Deep learning has shown to achieve dramatic results in image classification tasks. However, deep learning models require large amounts of data to train. Most of the real-world datasets, generally insect classification data does not have large number of training dataset. These images have a large amount of noise and various differences. The paper proposes a novel architectural model which removes the background noise and classify the Tiger beetles. Here object location is identified using contours by converting the original coloured image to white on black background. Then the remaining background is eliminated using grabcut algorithm. Later the extracted images are classified using a modified SqueezeNet transfer learning model to identify the tiger beetle class up to genus level. Transfer learning models with fewer trainable parameters performed well than the total number of parameters in the original model. When evaluating results it was identified that by freezing uppermost layers of SqueezeNet model better accuracy can be gained while freezing lowermost layers will reduce the validation accuracy. The proposed model achieved more than 90% for the test set in 40 epochs using 701,481 trainable parameters by freezing the top 19 layers of the original model. Improving the pre-processing to localize insect has improved the accuracy.

Item Type: Journal Article
Murdoch Affiliation(s): Information Technology, Mathematics and Statistics
Publisher: Elsevier
Copyright: © 2021 Elsevier B.V.
URI: http://researchrepository.murdoch.edu.au/id/eprint/60400
Item Control Page Item Control Page