Dried shiitake mushroom grade recognition using D-VGG network and machine vision

Frontiers in Nutrition

Nutrition

Li Wang¹  Qunfeng Niu¹  Penghao Dong¹  Kunming Jia¹  Qiao Wang² 

[1] School of Electrical Engineering, Henan University of Technology, Zhengzhou, China;School of Electrical Engineering, Henan University of Technology, Zhengzhou, China;Key Laboratory of Grain Information Processing and Control, Ministry of Education, Henan University of Technology, Zhengzhou, China;

关键词: dried shiitake mushroom;    grading;    image preprocessing;    deep learning;    classification model;   

DOI  :  10.3389/fnut.2023.1247075

 received in 2023-06-26, accepted in 2023-10-06,  发布年份 2023

来源: Frontiers

【 摘 要 】

Grading dried shiitake mushrooms is an indispensable production step, as there are large quality differences between different grades, which affect the product’s price and marketability. Dried shiitake mushroom samples have irregular shapes, small morphological differences between different grades of the same species, and they may occur in mixed grades, which causes challenges to the automatic grade recognition using machine vision. In this study, a comprehensive method to solve this problem is provided, including image acquisition, preprocessing, dataset creation, and grade recognition. The osprey optimization algorithm (OOA) is used to improve the computational efficiency of Otsu’s threshold binarization and obtain complete mushroom contours samples efficiently. Then, a method for dried shiitake mushroom grade recognition based on the improved VGG network (D-VGG) is proposed. The method uses the VGG16 network as the base framework, optimizes the convolutional layer of the network, and uses a global average pooling layer instead of a fully connected layer to reduce the risk of model overfitting. In addition, a residual module and batch normalization are introduced to enhance the learning effect of texture details, accelerate the convergence of the model, and improve the stability of the training process. An improved channel attention network is proposed to enhance the feature weights of different channels and improve the grading performance of the model. The experimental results show that the improved network model (D-VGG) can recognize different dried shiitake mushroom grades with high accuracy and recognition efficiency, achieving a final grading accuracy of 96.21%, with only 46.77 ms required to process a single image. The dried shiitake mushroom grade recognition method proposed in this study provides a new implementation approach for the dried shiitake mushroom quality grading process, as well as a reference for real-time grade recognition of other agricultural products.

【 授权许可】

Unknown   
Copyright © 2023 Wang, Dong, Wang, Jia and Niu.