基于CNN-DBN的小麦不完善粒识别技术研究

    Research on unsound wheat kernels recognition technology based on CNN-DBN

    • 摘要: 针对在实际应用场景下, 小麦不完善粒识别数据较少所产生识别率不佳的问题, 提出并实现了基于迁移学习的CNN-DBN小麦不完善粒识别方法。利用基于大型公开数据集ImageNet的预训练深度卷积神经网络(CNN)中的VGG-16、VGG-19和ResNet50进行小麦特征提取, 将获取的特征加以融合并输送至深度信念网络(DBN)进行分类。结果表明:CNN和DBN结合的方法用于小麦不完善粒识别,其中迁移学习VGG-16+VGG-19+ResNet50-DBN模型性能最好,其测试准确率可达91.86%;CNN-DBN模型既避免了小麦复杂的特征提取步骤,又使不完善粒识别因数据集规模小而导致识别率不理想的问题得到了改善;特征融合的方法使提取到的小麦图像信息更加丰富、全面。CNN-DBN模型结合了有监督网络和无监督网络的优点,对高维数据有更好的分类能力,为小麦不完善粒识别提供了理论支持。

       

      Abstract: The recognition of unsound wheat kernels is an important part of wheat quality inspection, and it is also a key indicator to measure wheat quality. Research on the recognition of unsound wheat kernels is of great significance to the correct evaluation of wheat quality. In recent years, researches on the recognition of unsound wheat kernels were mainly to directly optimize classical classification networks, and the problem of unsatisfactory recognition effect was often caused by insufficient wheat training dataset. Aiming at the problem of poor recognition rate due to insufficient recognition data of unsound wheat kernels in practical application scenarios, we designed and implemented a method for identifying unsound wheat kernels based on transfer learning CNN-DBN model. We used a jitter-type automatic feeding tray, a transparent carrier plate and a pair of high-definition industrial cameras to form an automatic wheat image acquisition device to obtain high-definition RGB images of wheat. Using the deep convolutional neural networks VGG-16, VGG-19 and ResNet50 pre-trained on the ImageNet dataset as feature extractors, the extracted wheat features were fused, and then classified by the deep belief network (DBN). We conducted experiments of single model, two-model fusion and three-model fusion, based on the performance of the three types of fusion models on the dataset, and finally chose the three-model fusion scheme with the best performance on the dataset. Meanwhile, the recognition results of each model using the softmax classifier and DBN as the classifier were compared. Finally, the transfer learning VGG-16+VGG-19 +ResNet50-DBN model was selected as the model for this task. Experiments showed that the transfer learning VGG-16+VGG-19+ResNet50-DBN model was used to identify unsound wheat kernels, and its accuracy can reach 91.86%. The pre-trained models we use were trained on the large-scale dataset ImageNet. Because the scale of the ImageNet dataset was very large, these pre-trained models also had good generalization performance for images outside the ImageNet dataset. Therefore, although our dataset was small, using the pre-trained models of the ImageNet dataset as the wheat feature extractor can still achieve a good recognition result. The proposed method not only avoided complicated feature extraction steps, but also improved the problem of unsatisfactory recognition rate due to the small size of the dataset in the recognition of unsound wheat kernels. The feature fusion method made the extracted wheat image information more abundant and comprehensive. The CNN-DBN model combined the advantages of supervised and unsupervised method, and had better classification capabilities for high-dimensional data. This work provides a theoretical support for the recognition of unsound wheat kernels.

       

    /

    返回文章
    返回