基于卷积神经网络与迁移学习的油茶病害图像识别  

Image recognition of Camellia oleifera diseases based on convolutional neural network & transfer learning

作　　者：龙满生[1,2] 欧阳春娟[1,2] 刘欢[1,2] 付青 Long Mansheng;Ouyang Chunjuan;Liu Huan;Fu Qing(School of Electronics and Information Engineering,Jinggangshan University,Ji’an 343009,China;Key Laboratory of Watershed Ecology and Geographical Environment Monitoring of NASG,Ji’an 343009,China)

机构地区：[1]井冈山大学电子与信息工程学院,吉安343009 [2]流域生态与地理环境监测国家测绘地理信息局重点实验室,吉安343009

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

基　　金：国家自然科学基金项目(41561065);流域生态与地理环境监测国家测绘地理信息局重点实验室开放基金课题(WE2015002);江西省自然科学基金项目(20151BAB203038;20161BAB204172)

摘　　要：传统的植物病害图像识别准确率严重依赖于耗时费力的人工特征设计。该文利用深度卷积神经网络强大的特征学习和特征表达能力来自动学习油茶病害特征,并借助迁移学习方法将Alex Net模型在Image Net图像数据集上学习得到的知识迁移到油茶病害识别任务。对油茶叶片图像进行阈值分割、旋转对齐、尺度缩放等预处理后,按照病害特征由人工分为藻斑病、软腐病、煤污病、黄化病和健康叶5个类别。每个类别各选取750幅图像组成样本集,从样本集中随机选择80%的样本用作训练集,剩余20%用作测试集。利用随机裁剪、旋转变换和透视变换对训练集进行数据扩充,以模拟图像采集的不同视角和减少网络模型的过拟合。在Tensor Flow深度学习框架下,基于数据扩充前后的样本集,对Alex Net进行全新学习和迁移学习。试验结果表明,迁移学习能够明显提高模型的收敛速度和分类性能;数据扩充有助于增加数据的多样性,避免出现过拟合现象;在迁移学习和数据扩充方式下的分类准确率高达96.53%,对藻斑病、软腐病、煤污病、黄化病、健康叶5类病害的F1得分分别达到94.28%、94.67%、97.31%、98.34%和98.03%。该方法具有较高的识别准确率,对平移、旋转具有较强的鲁棒性,可为植物叶片病害智能诊断提供参考。Leaf diseases are a serious problem in Camellia oleifera production.The occurrence of Camellia oleifera disease is affected by various factors,such as variety,cultivation environment,climate condition and management level.The key to effective prevention and cure of Camellia oleifera disease is to identify the disease type timely and accurately.Traditional computer vision methods for plant leaf disease recognition depend heavily on time-consuming and elaborate feature design.To solve this problem,a recognition model of Camellia olerfera leaf diseases based on convolutional neural network was proposed and transfer learning was used to improve model’s performance.Deep convolutional neural network has powerful capacities of feature learning and feature expression,which was used to learn features of diseased Camellia oleifera leaves.Transfer learning method was used to transfer the knowledge learned from ImageNet dataset by AlexNet to the identification task of Camellia oleifera diseases.The proposed model was implemented with Python programming language under the deep learning framework of Tensorflow by modifying the output number of the last fully connected layers in AlexNet to 5.We collected Camellia oleifera leaves in artificial Camellia oleifera land and took photos by mobile phone in bright indoor environment after flattening leaves.Leaf images were first converted from RGB(red,green,blue)color space to HSI(hue,saturation,intensity)color space,and then background was removed by threshold segmentation on hue and saturation channels.After segmentation,morphological open and close operations with a radius of 3 pixels were performed to remove burrs,holes and other noises,and thus the leaf mask was obtained by filling holes.Leaf mask was multiplied with the original image to obtain the colored leaf region.The colored leaf region was then rotated according to its principal axis angle and aligned horizontally.Based on the long edge,leaf image was scaled to 256×256 pixels.After these pretreatments,Camellia oleifera l

关 键 词：病害 分类 作物 油茶病害 图像识别 深度学习 迁移学习 

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