A New Deep Learning Model for the Classification of Poisonous and Edible Mushrooms Based on Improved AlexNet Convolutional Neural Network

Abstract

The difficulty involved in distinguishing between edible and poisonous mushrooms stems from their similar appearances. In this study, we attempted to classify five common species of poisonous and edible mushrooms found in Thailand, Inocybe rimosa, Amanita phalloides, Amanita citrina, Russula delica, and Phaeogyroporus portentosus, using the convolutional neural network (CNN) and region convolutional neural network (R-CNN). This study was motivated by the yearly death toll from eating poisonous mushrooms in Thailand. In this research, a method for the classification of edible and poisonous mushrooms was proposed and the testing time and accuracy of three pretrained models, AlexNet, ResNet-50, and GoogLeNet, were compared. The proposed model was found to reduce the duration required for training and testing while retaining a high level of accuracy. In the mushroom classification experiments using CNN and R-CNN, the proposed model demonstrated accuracy levels of 98.50% and 95.50%, respectively.