Laser & Optoelectronics Progress, Volume. 60, Issue 20, 2010003(2023)
Hyperspectral Imaging-Based Quality Classification for Kiwifruit by Incorporating Three-Dimensional Convolution Neural Network and Haar Wavelet Filter
To address challenges in the non-destructive inspection and classification of kiwifruit hardness quality, we propose a classification model that incorporates hyperspectral imaging technology and a convolution neural network. This network combines the spatial feature information extracted by the Haar wavelet and the space-spectrum joint information extracted by the three-dimensional (3D) convolution kernel. In this network, the data decomposition of channel connections is executed to ensure that all features can be utilized by the model, which improves the ability of network feature learning. Experiments on the acquired hyperspectral image-based, self-made kiwifruit hardness quality dataset (named Kiwi_seed) demonstrate that the Haar wavelet transform module can significantly improve the feature extraction ability of the network. Ablation experiments reveal that the classification accuracy of the model incorporating the Haar wavelet transform module is increased by 7.4% and reaches the optimum level at 97.3%, which is better than the classical image classification network. The proposed classification model can be effectively used for the non-destructive inspection and classification of kiwifruit quality.
Get Citation
Copy Citation Text
Ke Jin, Zhiqiang Guo, Yunliu Zeng, Gang Ding. Hyperspectral Imaging-Based Quality Classification for Kiwifruit by Incorporating Three-Dimensional Convolution Neural Network and Haar Wavelet Filter[J]. Laser & Optoelectronics Progress, 2023, 60(20): 2010003
Category: Image Processing
Received: Nov. 23, 2022
Accepted: Dec. 15, 2022
Published Online: Sep. 28, 2023
The Author Email: Guo Zhiqiang (guozhiqiang@whut.edu.cn), Zeng Yunliu (zengyl@mail.hzau.edu.cn)