Deep learning algorithm development for early detection of Botrytis cinerea infected strawberry fruit using hyperspectral fluorescence imaging

Abstract

Botrytis cinerea is a strawberry disease that causes economic loss worldwide. If a disease outbreak occurs during storage or transportation, it can spread rapidly to neighboring objects; thus, there is a need to develop early diagnostic techniques to prevent it. In this study, we developed a method to rapidly and nondestructively determine the infection stage in strawberry fruit using hyperspectral fluorescence imaging. ‘Keumsil’ cultivar strawberries were used, and hyperspectral fluorescence images were acquired over 144 h in control and inoculation groups. Strawberries were categorized into four infection stages based on visible mold spores: healthy, asymptomatic, infected, and after-infected. Hyperspectral fluorescence spectra were extracted to develop a one-dimensional convolutional neural network (1D-CNN) model based on partial least squares-discriminant analysis (PLS-DA), VGG-19, and ResNet-50; data augmentation techniques and six spectral preprocessing techniques were applied to the datasets. The application of data augmentation techniques improved the performances of the PLS-DA and 1D-CNN models in determining the infection stage. The performance of the ResNet-50-based 1D-CNN model with mean normalization data and data augmentation technique was the best, with 96.88% precision, 96.87% recall, 96.85% F1-score, and 96.86% accuracy. The results of this study showed that it is possible to determine the infection stage of Botrytis cinerea on strawberry fruit using hyperspectral fluorescence imaging and 1D-CNN techniques. This technology is expected to be applied for the early detection of Botrytis cinerea in strawberry growth, postharvest sorting and packing, and distribution stages.