Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2022, Vol. 43 ›› Issue (4): 580-589.DOI: 10.11996/JG.j.2095-302X.2022040580

• Image Processing and Computer Vision • Previous Articles     Next Articles

Research and application of wild mushrooms classification based on multi-scale features to realize hyperparameter evolution

  

  1. 1. School of Information Engineering, Nanchang Institute of Technology, Nanchang Jiangxi 330000, China;
    2. School of Mechanical Engineering, Nanchang Institute of Technology, Nanchang Jiangxi 330000, China
  • Online:2022-08-31 Published:2022-08-15
  • Contact: HUANG Zhi-kai (1969), professor, Ph.D. His main research interest covers image processing, etc
  • About author:ZHANG Dun (1996), master student. His main research interests cover image processing and target detection
  • Supported by:
    National Key Research and Development Plan of China (2019YFB1704502); National Natural Science Foundation of China (61472173);
    The Grants from the Educational Commission of Jiangxi Province of China (GJJ151134)

Abstract:

In China, there are frequent poisoning events caused by ingestion of inedible wild mushrooms every summer, especially in Southwest China, such as Yunnan. This is due to the slight differences in inter-class characteristics of wild mushrooms and the complex image backgrounds in actual scenarios, making it difficult to distinguish only by naked eyes. At present, although there are many methods to classify wild mushrooms, and the most reliable way is molecular identification, the relevant techniques are time-consuming and require a high threshold, so they are not suitable for real-time classification and detection. To solve this problem, an approach based on deep learning was proposed. This approach employed the attention mechanism convolution block attention module (CBAM), was combined with multi-scale fusion, and added the anchor layer. The hyperparameter evolution idea was adopted to adjust the hyperparameter during the model training, so as to improve the recognition accuracy. Compared with standard target detection networks, such as SSD, Faster Rcnn, and Yolo series, the proposed model can classify and detect wild mushrooms more accurately. Compared with the original Yolov5, after the proposed model was improved, Map was improved by 3.7% and reached 93.2%, precision by 1.3%, Recall by 1.0%, and model detection speed by 2.3%. Compared with SSD, Map was improved by 14.3%. Finally, the model was simplified and deployed on Android devices to increase its practicability, thus solving the current problem of poisoning caused by eating inedible wild mushrooms because of the difficulty of identification.

Key words: computer application, convolutional neural network, multi scale features, hyperparameter evolution; attention mechanism, edible wild mushrooms, target detection 

CLC Number: