Transfer Learning for Wildlife Classification: Evaluating YOLOv8 against DenseNet, ResNet, and VGGNet on a Custom Dataset
PDF

Keywords

Convolutional Neural Network (CNN)
Endangered Species Detection
Image Classification
Transfer Learning
YOLO

How to Cite

Sharma, Subek, Sisir Dhakal, and Mansi Bhavsar. 2024. “Transfer Learning for Wildlife Classification: Evaluating YOLOv8 Against DenseNet, ResNet, and VGGNet on a Custom Dataset”. Journal of Artificial Intelligence and Capsule Networks 6 (4): 415-35. https://doi.org/10.36548/jaicn.2024.4.003.

Abstract

This study evaluates the performance of various deep learning models, specifically DenseNet, ResNet, VGGNet, and YOLOv8, for wildlife species classification on a custom dataset. The dataset comprises 575 images of 23 endangered species sourced from reputable online repositories. The study utilizes transfer learning to fine-tune pre-trained models on the dataset, focusing on reducing training time and enhancing classification accuracy. The results demonstrate that YOLOv8 outperforms other models, achieving a training accuracy of 97.39% and a validation F1-score of 96.50‘%. These findings suggest that YOLOv8, with its advanced architecture and efficient feature extraction capabilities, holds great promise for automating wildlife monitoring and conservation efforts.

PDF

References

S. Jnawali, H. Baral, S. Lee, et al., “The status of nepal mammals: The national red list series, department of national parks and wildlife conservation kathmandu, nepal,” Preface by Simon M. Stuart Chair IUCN Species Survival Commission The Status of Nepal’s Mammals: The National Red List Series, vol. 4, 2011.

Y. LeCun, L. Bottou, Y. Bengio, and P. Haffner, “Gradient-based learning applied to document recognition,” Proceedings of the IEEE, vol. 86, no. 11, pp. 2278–2324, 1998.

O. Russakovsky, J. Deng, H. Su, et al., “Imagenet large scale visual recognition challenge,” International journal of computer vision, vol. 115, pp. 211–252, 2015.

C. Szegedy, W. Liu, Y. Jia, et al., “Going deeper with convolutions,” in Proceedings of the IEEE conference on computer vision and pattern recognition, USA. 2015, pp. 1–9.

A. Krizhevsky, I. Sutskever, and G. E. Hinton, “Imagenet classification with deep convolutional neural networks,” Advances in neural information processing systems, vol. 25, 2012.

S. J. Pan and Q. Yang, “A survey on transfer learning,” IEEE Transactions on knowledge and data engineering, vol. 22, no. 10, pp. 1345–1359, 2009.

G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, “Densely connected convolutional networks,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, pp. 4700–4708.

K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 770–778.

K. Simonyan and A. Zisserman, “Very deep convolutional networks for large-scale image recognition,” arXiv preprint arXiv:1409.1556, 2014.

J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You only look once: Unified, real-time object detection,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 779– 788.

N. K. El Abbadi and E. M. T. A. Alsaadi, “An automated vertebrate animals classification using deep convolution neural networks,” in 2020 International Conference on Computer Science and Software Engineering (CSASE), IEEE, 2020, pp. 72–77.

A. G. Villa, A. Salazar, and F. Vargas, “Towards automatic wild animal monitoring: Identification of animal species in camera-trap images using very deep convolutional neural networks,” Ecological informatics, vol. 41, pp. 24–32, 2017.

M. Ibraheam, F. Gebali, K. F. Li, and L. Sielecki, “Animal species recognition using deep learning,” in Advanced Information Networking and Applications: Proceedings of the 34th International Conference on Advanced Information Networking and Applications (AINA-2020), Springer, 2020, pp. 523–532.

J. Li, J. Chen, B. Sheng, et al., “Automatic detection and classification system of domestic waste via multimodel cascaded convolutional neural network,” IEEE Transactions on Industrial Informatics, vol. 18, no. 1, pp. 163–173, 2022. doi: 10.1109/TII.2021.3085669.

C.-A. Brust, T. Burghardt, M. Groenenberg, et al., “Towards automated visual monitoring of individual gorillas in the wild,” in Proceedings of the IEEE International Conference on Computer Vision Workshops, 2017, pp. 2820–2830.

S. Beery, G. Van Horn, and P. Perona, “Recognition in terra incognita,” in Proceedings of the European conference on computer vision (ECCV), 2018, pp. 456–473.

A. Yılmaz, G. N. Uzun, M. Z. G¨urb¨uz, and O. Kıvrak, “Detection and breed classification of cattle using yolo v4 algorithm,” in 2021 International Conference on INnovations in Intelligent SysTems and Applications (INISTA), IEEE, 2021, pp. 1–4.

M. Kumar, A. K. Yadav, M. Kumar, and D. Yadav, “Bird species classification from images using deep learning,” in International Conference on Computer Vision and Image Processing, Springer, 2022, pp. 388– 401.

H. Nguyen, S. J. Maclagan, T. D. Nguyen, et al., “Animal recognition and identification with deep convolutional neural networks for automated wildlife monitoring,” in 2017 IEEE international conference on data science and advanced Analytics (DSAA), IEEE, 2017, pp. 40–49.

iNaturalist, iNaturalist, https://inaturalist.org/, 2022.

ZooChat, ZooChat: The world’s largest community of zoo and animal conservation enthusiasts, https://www.zoochat.com/community/, 2002.

C. Caldas, split-folders (version 0.5.1), Available at: https://pypi.org/project/split- folders/, 2021.

M. A. A. Babu, S. K. Pandey, D. Durisic, A. C. Koppisetty, and M. Staron, “Impact of image data splitting on the performance of automotive perception systems,” in International Conference on Software Quality, Springer, 2024, pp. 91–111.

C. Shorten and T. M. Khoshgoftaar, “A survey on image data augmentation for deep learning,” Journal of big data, vol. 6, no. 1, pp. 1–48, 2019.

J. Lu, L. Tan, and H. Jiang, “Review on convolutional neural network (cnn) applied to plant leaf disease classification,” Agriculture, vol. 11, no. 8, p. 707, 2021.

S. Sharma, R. Ghimire, S. Gurung, and S. GC, “Gantavya: A landmark recognition system,” 2024.

V. H. Phung and E. J. Rhee, “A deep learning approach for classification of cloud image patches on small datasets,” Journal of information and communication convergence engineering, vol. 16, no. 3, pp. 173–178, 2018.

D. Reis, J. Kupec, J. Hong, and A. Daoudi, “Real-time flying object detection with yolov8,” arXiv preprint arXiv:2305.09972, 2023.

C.-Y. Wang, H.-Y. M. Liao, Y.-H. Wu, P.-Y. Chen, J.-W. Hsieh, and I.-H. Yeh, “Cspnet: A new backbone that can enhance learning capability of cnn,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops, 2020, pp. 390–391.

S. Liu, L. Qi, H. Qin, J. Shi, and J. Jia, “Path aggregation network for instance segmentation,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, pp. 8759–8768.