食品科学 ›› 2017, Vol. 38 ›› Issue (24): 283-287.doi: 10.7506/spkx1002-6630-201724046

• 安全检测 • 上一篇    下一篇

基于CNN神经网络的小麦不完善粒高光谱检测

于重重,周兰,王鑫,吴静珠,刘倩   

  1. (北京工商大学计算机与信息工程学院,食品安全大数据技术北京市重点实验室,北京 100048)
  • 出版日期:2017-12-25 发布日期:2017-12-07
  • 基金资助:
    土壤植物机器系统技术国家重点实验室开放课题(2014-SKL-05);北京工商大学两科基金培育项目(LKJJ2015-22)

Hyperspectral Detection of Unsound Kernels of Wheat Based on Convolutional Neural Network

YU Chongchong, ZHOU Lan, WANG Xin, WU Jingzhu, LIU Qian   

  1. (Beijing Key Laboratory of Big Data Technology for Food Safety, School of Computer and Information Engineering, Beijing Technology and Business University, Beijing 100048, China)
  • Online:2017-12-25 Published:2017-12-07

摘要: 利用高光谱成像技术对小麦不完善粒进行无损检测。以932?个小麦为样本,其中正常粒样本486?个、破损粒样本170?个、虫蚀粒样本149?个及黑胚粒样本127?个为研究对象,通过高光谱图像采集系统采集样本的光谱信息,然后从每个样本的116?个波段中选取30?个波段,建立基于深度学习的卷积神经网络(convolutional neural networks,CNN)模型。实验中的CNN采用2?个卷积层,第1层采用大小为3×3的32?个卷积核,第2层采用大小为5×5的64?个卷积核,池化层采用最大池,激活函数采用修正线性单元,为避免过拟合,在全连接层后面接入dropout层,参数设置为0.5,其他卷积参数均为默认值,得到校正集总识别率为100.00%,测试集总识别率为99.98%。最后,以支持向量机(support vector machine,SVM)为基线模型进行对比,从116?个波段中选取90?个波段进行建模,测试集总识别率为94.73%。通过实验对比可以看出,CNN模型比SVM模型识别率高。研究表明CNN模型能够实现对小麦不完善粒的准确、快速、无损检测。

关键词: 小麦, 不完善粒, 高光谱检测, 卷积神经网络模型

Abstract: A nondestructive examination was conducted to identify unsound kernels by using hyperspectral imaging technology. Based on 932 wheat samples including 486 normal samples, 170 damaged samples, 149 worm-eaten samples and 127 black germ kernel samples, a hyperspectral image acquisition system was used for collecting hyperspectral information, and then a convolutional neural network (CNN) was established based on 30 wavebands selected from 116 wavebands for each sample. The CNN model comprised two convolution layers. The first layer consisted of 32 convolution kernels (3 × 3) and the second layer consisted of 64 convolution kernels (5 × 5). The pooling layer was developed with the maximum pool. The activation function was developed with rectified linear units (ReLu). To avoid overfitting, a dropout layer was linked to the fully connected layer, and the parameter was set as 0.5. When other parameters remained default, recognition rates for the calibration and test sets were 100.00% and 99.98% respectively. Finally, a support vector machine (SVM) model was built and compared with the CNN model. The SVM model developed with 90 wavebands selected from 116 wavebands showed a recognition rate of 94.73% for the test set. The recognition rate of the CNN model was better than that of the SVM model. Thus, this research showed that the CNN model allowed for accurate, rapid and nondestructive detection of unsound wheat kernels.

Key words: wheat, unsound kernels, hyperspectral detection, CNN model

中图分类号: