基于卷积神经网络的花生籽粒完整性识别算法及应用  

Identification algorithm and application of peanut kernel integrity based on convolution neural network

作　　者：赵志衡[1] 宋欢[1] 朱江波 卢雷[1] 孙磊 Zhao Zhiheng;Song Huan;Zhu Jiangbo;Lu Lei;Sun Lei(School of Electrical Engineering&Automation,Harbin Institute of Technology,Harbin 150001,China;Shanghai Anzai Manufacturing Co.,Ltd.,Shanghai 201109,China)

机构地区：[1]哈尔滨工业大学电气工程及自动化学院,哈尔滨150001 [2]上海安西机械制造有限公司,上海201109

出　　处：《农业工程学报》2018年第21期195-201,共7页Transactions of the Chinese Society of Agricultural Engineering

基　　金：国家科技重大专项(2014zx04001171)

摘　　要：针对现有色选设备在花生颗粒筛选过程中处理速度慢、准确率低的缺点,提出基于卷积神经网络的花生籽粒完整性识别算法。以完好花生、表皮破损花生和果仁破损花生的分类为例,构建花生图像库;搭造卷积神经网络,提取花生图像特征;为提高分类准确率和实时性,从训练集构成、减小过拟合、加快训练收敛速度、简化网络结构等几方面对卷积神经网络进行优化;最终利用含2个卷积层、2个池化层、2个全连接层的3层神经网络实现了上述3类花生的分类。试验结果表明:该方法对花生分类的准确率达到98.18%,平均检测一幅单粒花生图像的时间为18ms,与现有色选设备相比有效提高了色选设备筛选的准确率和实时性。Aiming at the shortcomings of the existing color sorter machine for crop sorting,such as slow processing speed,low accuracy,and the dependence on experience value,a granular crop integrity identification algorithm based on convolutional neural network was proposed.Taking the classification of intact peanuts,skin damaged peanuts and half peanuts as instance,the three types of peanut images were acquired.After comparing the filtering effects of mean filtering,median filtering and Gaussian filtering,median filtering was adopted for image preprocessing.407 effective peanut images were divided into the above three categories and manually labeled.Then the images were divided into training sets and validation sets,and the above three types of peanut pictures in the training set and the validation set were evenly distributed.A convolutional neural network with 4 convolutional layers,4 pooling layers and 3 fully connected layers was built to extract the peanut image features.The accuracy of testing peanut classification on the CPU(central processing unit)platform combined GPU(graphics processing unit)was 90.91%.In contrast,the classification accuracy of the traditional BP neural network was 85.45%.It could be seen that the convolutional neural network algorithm constructed in this paper effectively improved the accuracy of granular crop recognition.In order to further improve the accuracy and real-time performance of the classification algorithm,it was necessary to optimize the established convolutional neural network.Over-fitting referred to the fact that when a model was overly complex,it could"memorize"the portion of random noise in each training data and forgot to"learn"the tendencyof the training data.In this paper,the regularization method was used to reduce the over-fitting,and the experimental results of L1 regularization and L2 regularization were compared.It was proved that the L2 regularization on the data set effectively improved the classification accuracy and reduced the over-fitting.In the process of traini

关 键 词：农产品 图像处理 识别 卷积神经网络 特征提取 色选系统 花生颗粒筛选 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]