CNN Transfer Learning for Automatic Fruit Recognition for Future Class of Fruit

  • Israr Hussain College of Electronic and Information Engineering, Shenzhen University, Shenzhen China
  • Shunquan Tan College of Electronic and Information Engineering, Shenzhen University, Shenzhen China
  • Wajid Ali Department of Mathematical Sciences Karakoram International University Gilgit Pakistan
  • Amjad Ali Department of Agriculture and Food Technology Karakoram International University Gilgit Pakistan
Keywords: Deep Learning, Fruit Recognition, Transfer Learning, CNN

Abstract

Deep fruit recognition model learned on big dataset outperform fruit recognition task on difficult unconstrained fruit dataset. But in practice, we are often lack of resources to learn such a complex model, or we only have very limited training samples for a specific fruit recognition task. In this study we address the problem of adding new classes to an existing deep convolutional neural network framework. We extended our prior work for automatic fruit recognition by applying transfer learning techniques to adding new classes to existing model which was trained for 15 different kind of fruits. Pre-trained model was previously trained on a large-scale dataset of 44406 images. To add new class of fruit in our pre-trained model, we need to train a new classifier which will be trained for scratch, on the top of pre-trained model so, that we can re- purpose the feature learned previously for the dataset. Transfer learning using our pre-trained model has been demonstrated to give the best classification accuracy of 95.00%. The experimental results demonstrate that our proposed CNN framework is superior to the previous state-of-the- art networks.

References

. A. Rocha, D. C. Hauagge, J. Wainer, S. Goldenstein ‘Automatic fruit and vegetable classification from images’, Computers and Electronics in Agriculture, 2010, 70, pp. 96– 104.

. A. Baltazar, J. I. Aranda, G. Gonzalez Aguilar, Bayesian classification of ripening stages of tomato fruit using acoustic impact and colorimeter sensor data, Computers and electronics in agriculture’, 2008, 60, pp. 113–121.

. S. Pouyanfar, S. Sadiq, Y. Yan, H. Tian, Y. Tao, M. P. Reyes, M.-L. Shyu, S.-C. Chen, S. Iyengar, ‘A survey on deep learning: Algorithms, techniques, and applications, ACM Computing Surveys (CSUR)’, 2018, 51, pp.51, 1–36

. Hussain, Israr, Qianhua He and Zhuliang Chen. ‘Automatic Fruit Recognition Based on DCNN for Commercial Source Trace System. International Journal of Computer Science & Applications’, 2018, 8, pp: 01-14.

. I. Hussain, W. L. Wu, H. Q. Hua, N. Hussain, ‘Intra-class recognition of fruits using dcnn for commercial trace back-system, in: Proceedings of the 4th International Conference on Multimedia Systems and Signal Processing (MSSP)’, 2019, pp. 194–199.

. Mure¸San, Horea, and Mihai Oltean. ‘Fruit recognition from images using deep learning’, Acta Universitatis Sapientiae Informatica’, 2018, 10, pp. 26-42.

. Wan, Shaohua, and Sotirios Goudos. ‘Faster R-CNN for multi-class fruit detection using a robotic vision system’, Computer Networks, 2010, 168, pp. 107036.

. Ludwig, I. and Ludwig A. W. W ‘Kondo effect induced by a magnetic field’, Phys. Rev. B, 2001, 64, pp. 045328

. Israr Hussain, Qianhua He, Zhuliang Chen, & Wei Xie. (2018). Fruit Recognition dataset (Version V 1.0) [Data set]. Zenodo. http://doi.org/10.5281/zenodo.1310165

. Alvarez-Canchila, O. I., D. E. Arroyo-Pérez, A. Patino-Saucedo, H. Rostro González, and A. Patino-Vanegas. ‘Colombian fruit and vegetables recognition using convolutional neural networks and transfer learning’, In Journal of Physics: Conference Series, 2020., vol. 1547, no. 1, p. 012020. IOP Publishing,

. Siddiqi, Raheel. ‘Effectiveness of Transfer Learning and Fine Tuning in Automated Fruit Image Classification’, In Proceedings of the 2019 3rd International Conference on Deep Learning Technologies, 2019 pp. 91-100.

. Tan, Chuanqi, Fuchun Sun, Tao Kong, Wenchang Zhang, Chao Yang, and Chunfang Liu. ‘A survey on deep transfer learning’, In International conference on artificial neural networks, 2018, pp. 270-279.

Published
2020-10-08
How to Cite
Hussain, I., Tan, S., Ali, W., & Ali, A. (2020). CNN Transfer Learning for Automatic Fruit Recognition for Future Class of Fruit. International Journal of Computer (IJC), 39(1), 88-96. Retrieved from https://ijcjournal.org/index.php/InternationalJournalOfComputer/article/view/1824
Section
Articles