Vol.30, No.3, 2021, pp.995-1005, doi:10.32604/iasc.2021.019020
OPEN ACCESS
ARTICLE
Predicting the Breed of Dogs and Cats with Fine-Tuned Keras Applications
  • I.-Hung Wang1, Mahardi2, Kuang-Chyi Lee2,*, Shinn-Liang Chang1
1 Department of Power Mechanical Engineering, National Formosa University
2 Department of Automation Engineering, National Formosa University, Huwei Township, Taiwan
* Corresponding Author: Kuang-Chyi Lee. Email:
(This article belongs to this Special Issue: Machine Learning and Deep Learning for Transportation)
Received 29 March 2021; Accepted 30 April 2021; Issue published 20 August 2021
Abstract
The images classification is one of the most common applications of deep learning. Images of dogs and cats are mostly used as examples for image classification models, as they are relatively easy for the human eyes to recognize. However, classifying the breed of a dog or a cat has its own complexity. In this paper, a fine-tuned pre-trained model of a Keras’ application was built with a new dataset of dogs and cats to predict the breed of identified dogs or cats. Keras applications are deep learning models, which have been previously trained with general image datasets from ImageNet. In this paper, the ResNet-152 v2, Inception-ResNet v2, and Xception models, adopted from Keras application, are retrained to predict the breed among the 21 classes of dogs and cats. Our results indicate that the Xception model has produced the highest prediction accuracy. The training accuracy is 99.49%, the validation accuracy is 99.21%, and the testing accuracy is 91.24%. Besides, the training time is about 14 hours and the predicting time is about 18.41 seconds.
Keywords
Image classification; deep learning; Keras; Inception; ResNet; Xception
Cite This Article
Wang, I., , M., Lee, K., Chang, S. (2021). Predicting the Breed of Dogs and Cats with Fine-Tuned Keras Applications. Intelligent Automation & Soft Computing, 30(3), 995–1005.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.