Classification of soybean seeds based on RGB reconstruction of hyperspectral images

Abstract

Soyabean is an incredibly significant component of Chinese agricultural product, and categorizing soyabean seeds allows for a better understanding of the features, attributes, and applications of many species of soyabean. This enables farmers to choose appropriate seeds for sowing in order to increase production and quality. As a result, this thesis provides a method for classifying soybean seeds that uses hyperspectral RGB picture reconstruction. Firstly, hyperspectral images of seven varieties of soybean, H1, H2, H3, H4, H5, H6 and H7, were collected by hyperspectral imager, and by using the principle of the three base colours, the R, G and B bands which have more characteristic information are selected to reconstruct the images with different texture and colour characteristics to generate a new dataset for seed segmentation, and finally, a comparison is made with the classification effect of the seven models. The experimental results in ResNet34 show that the classification accuracy of the dataset before and after RGB reconstruction increases from 88.87% to 91.75%, demonstrating that RGB image reconstruction can strengthen image features; ResNet18, ResNet34, ResNet50, ResNet101, CBAM-ResNet34, SENet-ResNet34, and SENet-ResNet34-DCN models have classification accuracies of 72.25%, 91.75%, 89%, 88.48%, 92.28%, 92.80%, and 94.24%, respectively.SENet-ResNet34-DCN achieves the greatest classification accuracy results, with a model loss of roughly 0.3. The proposed SENet-ResNet34-DCN model is the most effective at classifying soybean seeds. By classifying and optimally selecting seed varieties, agricultural production can become more scientific, efficient, and sustainable, resulting in higher returns for farmers and contributing to global food security and sustainable development.