摘要
深度学习在玉米病害识别领域应用广泛并取得了较好的效果,但存在低分辨率条件下训练效果差的问题,本文提出一种基于超分辨率生成对抗网络的玉米病害分类识别模型。为了实现低分辨率玉米病斑图像到高分辨率图像的恢复,提出基于双注意力机制的增强型超分辨率生成对抗网络模型,该模型生成的高分辨率重建图像与其他超分辨率图像重建模型相比,峰值信噪比(Peak signal-to-noise ratio,PSNR)和结构相似性指数(Structural similarity index measure,SSIM)平均提升2.1 dB和0.049;与4种不同的分类网络的结合,准确率均高于低分辨率图像,平均提升28.1个百分点。在模型对比及消融可视化实验中,模型识别玉米病斑准确率平均超出其它模型1.3个百分点,精确率达到97.8%。实验结果表明,双注意力机制的加入和损失函数的改变增加了模型对高频特征的恢复能力和稳健性,提高了玉米叶片病斑分类识别率,可为农作物定点监测或无人机田间监测中低分辨率叶片病害图像的精准识别提供参考。
Maize is one of the most important food crops in China.Leaf diseases of maize can seriously damage its yield,so the correct identification of disease is of great significance.However,the efficiency of traditional manual identification of leaf diseases is low.The resolution of disease images collected from agricultural fixed points or drone monitoring is low,and the key features are not significant,which cannot meet the image resolution requirements of classification and recognition models.The training effect is poor,making it difficult to accurately identify leaf diseases.To this end,a maize disease classification and recognition model based on an improved super-resolution generative adversarial network(SRGR)was designed.The images of maize leaf disease were divided into four types:large spot,rust,gray spot,and healthy leaves.The data set was divided into low resolution(LR)and high resolution(HR)images that corresponded one-to-one.In order to realize the restoration of low-resolution maize spot images to high-resolution images,this model proposed an improved strategy for the enhanced super-resolution generative adversarial networks(ESRGAN)model based on dual attention mechanism.LR images were input into the highfrequency feature reconstruction network,and channel attention(CA)mechanism after each residual dense block(RRDB)was added to extract deep detailed features of the image,making the model highly targeted in reconstructing highfrequency details and reducing the possibility of pseudo texture phenomenon.The generation network was divided into encoding and decoding parts,and the spatial attention mechanism was introduced into U-shaped dense block with skip layers to maximize the retention of maize disease effective features in the middle and low levels of the LR image of maize lesions.The probability value of high-frequency features in the input feature map was calculated to determine the position of reconstructed lesion features in the image.The WGAN-GP loss function was used to train the network to solve the
作者
马铁民
曲浩
高雅
王雪
MA Tiemin;QU Hao;GAO Ya;WANG Xue(College of Information Technology,Heilongjiang Bayi Agricultural University,Daqing 163319,China)
出处
《农业机械学报》
EI
CAS
CSCD
北大核心
2024年第11期49-56,67,共9页
Transactions of the Chinese Society for Agricultural Machinery
基金
国家自然科学基金项目(52275246)
黑龙江八一农垦大学三横三纵支持计划项目(ZRCPY202018)
黑龙江八一农垦大学人才引进科研启动项目(XDB202115)。
关键词
玉米叶片
病害识别
超分辨率图像重建
生成对抗网络
maize leaf
disease identification
super-resolution image reconstruction
generative adversarial network
