基于DXNet模型的富士苹果外部品质分级方法研究
External Quality Grading Method of Fuji Apple Based on Deep Learning
-
摘要: 针对传统计算机视觉技术在苹果外部品质分级中准确率较低、鲁棒性较差等问题,提出了基于深度学习的苹果外观分级方法(多卷积神经网络融合DXNet模型)。首先,在延安市超市、果园等场所实地拍摄不同外观等级的苹果图像15 000幅,并进行人工标记,建立了外部品质信息覆盖度广、样本量大的苹果图像数据库;然后在对比分析经典卷积网络模型的基础上,采用模型融合的方式对经典模型进行优化改进,抽取经典模型卷积部分进行融合,作为特征提取器,共享全连接层用作分类器,并采用批归一化和正则化技术防止模型过拟合。试验评估采用15 000幅图像进行训练、4 500幅图像进行测试,结果表明,DXNet模型的分级准确率高于经典模型,分级准确率达到97.84%,验证了本文方法用于苹果外部品质分级的有效性。Abstract: The research and development of high-precision and low-cost apple intelligent grading technology is the core issue to extend the apple industrial chain and improve the quality and efficiency of the fruit industry.In order to solve the problems of low accuracy and weak robustness of traditional computer vision technology in apple external quality classification,an apple appearance classification method based on deep learning(multiple convolutional neural network DXNet model) was proposed.Firstly,totally 15 000 apple images covering different appearance levels were taken in Yan’an supermarkets,orchards and other places,and then labeled manually.A database of apple images with extensive coverage of external quality information and large sample size was established.Then,on the basis of comparing and analyzing the classical convolution network model,the classical model was optimized and improved by the method of model fusion,and the convolution part of the classical model was extracted and fused to be the feature extractor,and the fully connected layer was shared to be the classifier,batch normalization and regularization techniques were used to prevent the model from over fitting.Totally 15 000 images were used for training and 4 500 images were used for testing.The results showed that the classification accuracy of the improved DXNet model was higher than that of the classical model,and the classification accuracy reached 97.84%,the validity of the method applied to apple external quality classification was verified.