TITLE
DETECTION OF UNHEALTHY FRUITS USING IMAGE CLASSIFICATION BASED ON A DEEP TRANSFER LEARNING FRAMEWORK
AUTHOR(S)
Parama Bagchi1, Dušan Marković2, Zoran Stamenkovic3, Dijana Stojić4*, Debotosh Bhattacharjee5
ABSTRACT
This paper presents a methodology for detecting unhealthy lemons in a lemon dataset based on a deep Convo-lutional Neural Network (CNN) and transfer learning. Initially, a CNN model was developed and trained, after which a new model was framed on the existing CNN model using transfer learning. The base model was trained to a satisfying level of accuracy (of 98.55%). Afterwards, we incorporated this pre-trained model into a customized transfer learning framework. The final model was tested using a different augmented lemon dataset. In this case, no additional training was performed on the transfer learning model and the resultant model achieved an accuracy of 95.91 %.
DOI
http://www.doi.org/10.70456/KKFD9917
DOWNLOAD
https://unitechsp.tugab.bg/images/2024/4-CST/s4_p202_v2.pdf
How to cite this article:
Parama Bagchi1, Dušan Marković2, Zoran Stamenkovic3, Dijana Stojić4*, Debotosh Bhattacharjee5, DETECTION OF UNHEALTHY FRUITS USING IMAGE CLASSIFICATION BASED ON A DEEP TRANSFER LEARNING FRAMEWORK, UNITECH – SELECTED PAPERS - 2024