| 基于Enhanced VGG16的油茶品种分类 |
| |
| 引用本文： | 孟志超,贺磊盈,杜小强,张国凤,姚小华,吴顺凯,郭豪鉴.基于Enhanced VGG16的油茶品种分类[J].农业工程学报,2022,38(10):176-181. |
| |
| 作者姓名： | 孟志超 贺磊盈 杜小强 张国凤 姚小华 吴顺凯 郭豪鉴 |
| |
| 作者单位： | 1. 浙江理工大学机械与自动控制学院,杭州 310018;;1. 浙江理工大学机械与自动控制学院,杭州 310018;2. 浙江省种植装备技术重点实验室,杭州 310018;;3. 中国林业科学研究院亚热带林业研究所,杭州 311400 |
| |
| 基金项目： | 国家自然科学基金项目（31971798）;国家重点研发计划课题油茶生态经济型品种筛选及配套栽培技术（2019YFD1001602）;浙江省"领雁"研发攻关计划项目（2022C02057） |
| |
| 摘 要： | 随着油茶产业不断壮大,市场上也出现了油茶幼苗品系混乱、以假乱真、以次充好的现象,因此急需开发一种专门的分类识别算法实现不同油茶品种的准确识别。农业领域常用VGG、ResNet网络模型进行分类工作,但存在权重空间过大和准确率不高等问题。该研究对VGG16网络模型进行层间删减以及结构调整,提出了Enhanced VGG16网络模型,在油茶叶数据集上完成模型训练与测试,并与现有经典卷积神经网络（AlexNet、VGG16、Resnet50、InceptionV3、Xception）进行对比。结果表明,Enhanced VGG16网络模型的训练集准确率和测试集准确率分别为98.98%和98.44%,权重空间为90.6 MB。与原始VGG16模型相比,训练集准确率和测试集准确率分别提高3.08和2.05个百分点,权重空间下降165.4 MB,模型性能显著提升。Enhanced VGG16网络模型与经典卷积神经网络相对比,模型综合性能更优。该研究为通过油茶叶进行品种分类识别提供了依据,同时可为其他农作物品种识别提供参考。
|
| 关 键 词： | 深度学习 油茶叶 分类 Enhanced VGG16 hard-Swish ReLU6 |
| 收稿时间： | 2021/9/16 0:00:00 |
| 修稿时间： | 2022/5/11 0:00:00 |
Classification of Camellia oleifera based on Enhanced VGG16 network |
| |
| Meng Zhichao,He Leiying,Du Xiaoqiang,Zhang Guofeng,Yao Xiaohu,Wu Shunkai,Guo Haojian.Classification of Camellia oleifera based on Enhanced VGG16 network[J].Transactions of the Chinese Society of Agricultural Engineering,2022,38(10):176-181. |
| |
| Authors: | Meng Zhichao He Leiying Du Xiaoqiang Zhang Guofeng Yao Xiaohu Wu Shunkai Guo Haojian |
| |
| Institution: | 1. Faculty of Mechanical Engineering & Automation, Zhejiang Sci-Tech University, Hangzhou 310018, China;;1. Faculty of Mechanical Engineering & Automation, Zhejiang Sci-Tech University, Hangzhou 310018, China; 2. Key Laboratory of Transplanting Equipment and Technology of Zhejiang Province, Hangzhou 310018, China;;3. Research Institute of Subtropical Forestry, Chinese Academy of Forestry, Hangzhou 311400, China |
| |
| Abstract: | Abstract: Camellia oleifera is one of the four largest woody oil plants in the world. There are the largest planting area and yield among all woody oil plants, most of which are distributed in more than 1 100 counties and cities in 18 provinces of southern China. Various kinds of Camellia oleifera cultivars have emerged in the market, particularly with the continuous development of Camellia oleifera industry in recent years. Therefore, a rapid and accurate classification can be urgent to identify the different varieties of Camellia oleifera. Deep learning can also be expected to serve as a promising way for classification, due mainly to the strong performance in many operations, such as classification, detection, and segmentation. Although the VGG and ResNet network models have been commonly used for the classification in agriculture, some limitations still remain, such as too large weight space and low accuracy. In this study, a systematic classification was performed on the image of Camellia oleifera plant leaf using the enhanced VGG16 network. Four cultivars of Camellia oleifera were selected to test, including the Changlin No. 3, No. 4, No. 40, and No. 53 taken from the National Camellia oleifera Seed Base of Dongfanghong Forest Farm, Jinhua City, Zhejiang Province, China. An Epson Perfection V30 scanner was used to collect the data set of the Camellia oleifera leaf images with a clear texture. 1800 images were obtained for the four Camellia oleifera varieties each. The training set and test set were then divided into the proportion of 3:2, where the training set was 4 320, and the test set was 2 880. Some operations of data enhancement were performed on the image during training, such as brightness adjustment, and random enhancement. The Enhanced VGG16 network was constructed using the hard-Swish and ReLU6 activation function, Residual block, Dropout, L2 regularization, the inter-layer deletion, and structural adjustment for the VGG16 network model. The performance of Enhanced VGG16 network model was then evaluated to compare with the classical convolutional neural networks (AlexNet, VGG16, Resnet50, InceptionV3, Xception). The model was also trained and tested on the sampled Camellia oleifera data set. More importantly, the hyperparameters dominated the model training and performance. The test optimizer was selected as the RMSprop optimization, while the loss function was categorical_crossentropy, as well as the Batch Size and the learning rate were 4, and 0.000 01, respectively. Furthermore, the learning rate was reduced to 10% of the original, if the accuracy of two epochs training sets remained the constant, as the number of iterations increased. The epochs were set to 100, while the training stopped in advance, if the accuracy of the four epochs training set remained. The results show that the accuracy of the training and test set of the Enhanced VGG16 network model were 98.98% and 98.44%, respectively. The average detection time of a single image was 55.32 ms, and the space of weight was 90.6 MB. The accuracies of validation and test set were improved by 3.08 and 2.05 percentage points, respectively, compared with the original. The space of weight and the average detection time were reduced by 165.4 MB and 2.18 ms, respectively, indicating the better performance of the enhanced model. Additionally, the enhanced VGG16 network model performed better in the classification of the Camellia oleifera leaf, and was much easier to deploy in the mobile terminals and embedded devices, compared with AlexNet, VGG16, Resnet50, InceptionV3, and Xception networks. The Camellia oleifera leaf data set can be further expanded to enhance the images at different growth stages, in order to overcome the interference of light in the unstructured and field environment. This finding can also provide promising technical support for crop species identification. |
| |
| Keywords: | deep learning Camellia oleifera leaf classification Enhanced VGG16 hard-Swish ReLU6 |
|
| 点击此处可从《农业工程学报》浏览原始摘要信息 |
| 点击此处可从《农业工程学报》下载免费的PDF全文 |
|
