Towards Designing the Best Model for Classification of Fish Species Using Deep Neural Networks

Autor: Raajas Tongaonkar, Pranav Thorat, Vandana Jagtap
Rok vydání: 2020
Předmět:
Zdroj: Proceeding of International Conference on Computational Science and Applications ISBN: 9789811507892
Popis: Convolutional neural networks have become a powerful tool for classification since 2012. The winners of the ImageNet challenge have been neural networks for a long time now. Computer vision applications mostly resort to neural networks. This paper extends its application to classify fishes of 23 different species using VGGNet algorithm. The fish images used for training the network is obtained from a live video dataset. We implemented the traditional VGG-16 and proposed our scaled-down version of it, the VGG-8. The results of training these two algorithms were compared on the basis of micro average, macro average and weighted average. VGG-16 surpassed VGG-8 in almost all parameters but not by a large margin as proved in the results section. Smaller networks consume less memory and take lesser time for training as well as prediction. Hence, smaller networks can be used in simpler applications and can run on a less dedicated hardware setup having restrictions of memory and/or processing power. Thus, it forms a good comparison problem which has been addressed in the paper below.
Databáze: OpenAIRE