Comparison of 2D and 3D convolutional neural networks in hyperspectral image analysis of fruits applied to orange bruise detection

Abstract

Recent advances in hyperspectral imaging (HSI) have demonstrated its ability to detect defects in fruit that may not be visible in RGB images. HSIs can be considered 3D images containing two spatial dimensions and one spectral dimension. Therefore, the first question that arises is how to process this type of information, either using 2D or 3D models. In this study, HSI in the 550–900 nm spectral range was used to detect bruising in oranges. Sixty samples of Thompson oranges were subjected to a mechanical bruising process, and HSIs were taken at different time intervals: before bruising, and 8 and 16 h after bruising. The samples were then classified using two convolutional neural network (CNN) models, a shallow 7‐layer network (CNN‐7) and a deep 18‐layer network (CNN‐18). In addition, two different input processing approaches are used: using 2D information from each band, and using the full 3D data from each HSI. The 3D models were the most accurate, with 94% correct classification for 3D‐CNN‐18, compared to 90% for 3D‐CNN‐7, and less than 83% for the 2D models. Our study suggests that 3D HSI may be a more effective technique for detecting fruit bruising, allowing the development of a fast, accurate, and nondestructive method for fruit sorting.Practical ApplicationOrange bruises can reduce the market value of food, which is why the food processing industry needs to carry out quality inspections. An effective way to perform this inspection is by using hyperspectral images that can be processed with 2D or 3D models, either with deep or shallow neural networks. The results of the comparison performed in this work can be useful for the development of more accurate and efficient bruise detection methods for fruit inspection.