改进VGG模型在苹果外观分类中的应用

为了将采摘后的苹果进行外观分类,提出了一种基于卷积神经网络的方法,通过改进VGG卷积神经网络完成对外观正常苹果、病斑苹果和腐烂苹果的分类。在VGG-16网络的基础上,加入批归一化层、采用全局平均池化和联合损失函数的方法对其进行结构优化。在经过数据增广的数据集上,与其他分类方法进行对比,结果表明:改进后的VGG网络对外观正常苹果、病斑苹果和腐烂苹果的识别精度分别为99.61%、98.89%和99.26%,均高于未改进VGGNet、AlexNet和GoogLeNet算法,证明此网络能够很好地完成对苹果外观的分类识别,可为采摘后的苹果实现智能分类提供技术支持。

Application of Improved VGG Model in Apple Appearance Classification

In order to realize the classification of apples after picking, this paper proposes a method based on convolutional neural network to improve the classification of normal apples, diseased apples and rotten apples by improving VGG convolutional neural network. On the basis of the VGG-16 network, the batch normalization layer is added, and the global average pooling and joint loss function are used to optimize the structure. Compare with other classification methods on data-enhanced data sets. The results showed that the improved VGG network''s recognition accuracy for normal apple, diseased apple and rotten apple were 99.61%, 98.89% and 99.26%, respectively, which were higher than the unmodified VGGNet, AlexNet and GoogLeNet algorithms, which proved that the network can be very Good completion of the classification and recognition of the appearance of Apple, can provide technical support for the intelligent classification of Apple after picking.