Quality Grading Algorithm of Oudemansiella Raphanipes Based on Transfer Learning and MobileNetV2
Tongkai Li,Huamao Huang,Yangyang Peng,Hui Zhou,Haiying Hu,Ming Liu
DOI: https://doi.org/10.3390/horticulturae8121119
2022-01-01
Horticulturae
Abstract:As a traditional edible and medicinal fungus in China, Oudemansiella raphanipes has high economic benefits. In order to achieve the automatic classification of Oudemansiella raphanipes into four quality levels using their image dataset, a quality grading algorithm based on neural network models was proposed. At first, the transfer learning strategy and six typical convolution neural network models, e.g., VGG16, ResNet50, InceptionV3, NasNet-Mobile, EfficientNet, and MobileNetV2, were used to train the datasets. Experiments show that MobileNetV2 has good performance considering both testing accuracy and detection time. MobileNetV2 only needs 37.5 ms to classify an image, which is shorter by 11.76%, 28.57%, 46.42%, 59.45%, and 79.73%, respectively, compared with the classification times of InceptionV3, EfficientNetB0, ResNet50, NasNet-Mobile, and VGG16. Based on the original MobileNetV2 model, four optimization methods, including data augmentation, hyperparameter selecting, an overfitting control strategy, and a dynamic learning rate strategy, were adopted to improve the accuracy. The final classification accuracy can reach as high as 98.75%, while the detection time for one image is only 22.5 ms and the model size is only 16.48 MB. This quality grading algorithm based on an improved MobileNetV2 model is feasible and effective for Oudemansiella raphanipes, satisfying the needs in the production line.