Quality Grading Algorithm of Oudemansiella raphanipes Based on Transfer Learning and MobileNetV2

文献类型： 外文期刊

第一作者： Li, Tongkai

作者： Li, Tongkai;Peng, Yangyang;Zhou, Hui;Liu, Ming;Li, Tongkai;Huang, Huamao;Hu, Haiying

作者机构：

关键词： Oudemansiella raphanipes; quality grading; convolutional neural network; transfer learning

期刊名称：HORTICULTURAE （ 影响因子：2.923； 五年影响因子：3.582 ）

ISSN：

年卷期： 2022 年 8 卷 12 期

页码：

收录情况： SCI

摘要： 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.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序