Transfer Learning for Medicinal Plant Leaves Recognition: A Comparison with and without a Fine-Tuning Strategy
Vina Ayumi, Ermatita Ermatita, Abdiansah Abdiansah, Handrie Noprisson, Yuwan Jumaryadi, Mariana Purba, Marissa Utami, Erwin Dwika Putra
Abstract
Plant leaves are another common source of information for determining plant species. According to the dataset that has been collected, we propose transfer learning models VGG16, VGG19, and MobileNetV2 to examine the distinguishing features to identify medicinal plant leaves. We also improved algorithm using fine-tuning strategy and analyzed a comparison with and without a fine-tuning strategy to transfer learning models performance. Several protocols or steps were used to conduct this study, including data collection, data preparation, feature extraction, classification, and evaluation. The distribution of training and validation data is 80% for training data and 20% for validation data, with 1500 images of thirty species. The testing data consisted of a total of 43 images of 30 species. Each species class consists of 1-3 images. With a validation accuracy of 96.02 percent, MobileNetV2 with fine-tuning had the best validation accuracy. MobileNetV2 with fine-tuning also had the best testing accuracy of 81.82%.