基于CycleGAN-IA方法和M-ConvNext网络的苹果叶片病害图像识别
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(62203344)、陕西省科技厅自然科学基础研究重点项目(2022JZ-35)和国家级大学生创新创业计划项目(202210709012)


Image Recognition of Apple Leaf Disease Based on CycleGAN-IA Method and M-ConvNext Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对苹果叶片病害图像识别存在数据集获取困难、样本不足、识别准确率低等问题,提出基于多尺度特征提取的病害识别网络(Multi-scale feature extraction ConvNext, M-ConvNext)模型。采用一种结合改进的循环一致性生成对抗网络与仿射变换的数据增强方法(Improved CycleGAN and affine transformation, CycleGAN-IA),首先,使用较小感受野的卷积核和残差注意力模块优化CycleGAN网络结构,使用二值交叉熵损失函数代替CycleGAN网络的均方差损失函数,以此生成高质量样本图像,提高样本特征复杂度;然后,对生成图像进行仿射变换,提高数据样本的空间复杂度,该方法解决了数据样本不足的问题,用于辅助后续的病害识别模型。其次,构建M-ConvNext网络,该网络设计G-RFB模块获取并融合各个尺度的特征信息,GELU激活函数增强网络的特征表达能力,提高苹果叶片病害图像识别准确率。最后,实验结果表明,CycleGAN-IA数据增强方法可以对数据集起到良好的扩充作用,在常用网络上验证,增强后的数据集可以有效提高苹果叶片病害图像识别准确率;通过消融实验可得,M-ConvNex识别准确率可达9918%,较原ConvNext网络准确率提高0.41个百分点,较ResNet50、MobileNetV3和EfficientNetV2网络分别提高3.78、7.35、4.07个百分点,为后续农作物病害识别提供了新思路。

    Abstract:

    Aiming at the problems of difficult dataset acquisition, insufficient samples, and low recognition accuracy in apple leaf disease image recognition, a disease recognition network based on multi-scale feature extraction ConvNext (M-ConvNext) model was proposed. A data enhancement method combining improved CycleGAN and affine transformation (CycleGAN-IA) was used. Firstly, the CycleGAN network structure was optimized by using a convolutional kernel with a smaller sensory field and a residual attention module, and a binary cross-entropy loss function instead of the mean-variance loss function of CycleGAN network, in order to generate high-quality sample images and improve the complexity of sample features;then affine transformation was applied to the generated images to improve the spatial complexity of the data samples, which solved the problem of insufficient data samples, and was used to assist the subsequent disease recognition model. Secondly, the M-ConvNext network was constructed, which was designed with the G-RFB module to acquire and fuse the feature information of each scale, and the GELU activation function enhanced the feature expression ability of the network to improve the accuracy of apple leaf disease image recognition. Finally, the experimental results showed that the CycleGAN-IA data enhancement method can play a good role in expanding the dataset, and it was verified on the commonly used network that the enhanced dataset can effectively improve the accuracy of apple leaf disease image recognition;through the ablation and comparison experiments, the recognition accuracy of M-ConvNex can be up to 99.18%, which was 0.41 percentage points more than the original ConvNext network, and 3.78 percentage points, 7.35 percentage points, 4.07 percentage points higher than that of ResNet50, MobileNetV3, and EfficientNetV2 networks, respectively, which provided an idea and laid a foundation for the subsequent recognition of crop diseases.

    参考文献
    相似文献
    引证文献
引用本文

李云红,张蕾涛,李丽敏,苏雪平,谢蓉蓉,史含驰.基于CycleGAN-IA方法和M-ConvNext网络的苹果叶片病害图像识别[J].农业机械学报,2024,55(4):204-212. LI Yunhong, ZHANG Leitao, LI Limin, SU Xueping, XIE Rongrong, SHI Hanchi. Image Recognition of Apple Leaf Disease Based on CycleGAN-IA Method and M-ConvNext Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2024,55(4):204-212.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-09-13
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-10
  • 出版日期: