农业机械学报

北大核心,CA,JST,EI,CSCD

国内刊号:11-1964/S

国际刊号:1000-1298

农业机械学报杂志2024年第10期:基于改进生成对抗网络的甜樱桃数据增强方法

发布日期:

作者:韩翔,李玉强,高昂,马静怡,宫庆福,宋月鹏

单位:山东农业大学;山东省园艺机械与装备重点实验室,山东农业大学,山东农业大学,山东农业大学,山东农业大学,山东农业大学;山东省农业装备智能化工程实验室,

关键词:甜樱桃图像;数据生成;生成对抗网络;长尾类不平衡

基金:山东省现代农业产业技术体系果品产业创新团队项目(SDAIT-06-12)

为解决在数据不平衡条件下甜樱桃分类模型出现的长尾类不平衡问题,提出了一种基于深度卷积生成对抗网络(Deep convolutional generative adversarial networks,DCGAN)的缺陷甜樱桃图像增强方法。首先,在生成器部分引入多尺度残差块(MSRB)和CBAM注意力机制,增强了模型特征表达能力和生成图像细节质量,同时改善了梯度流;在判别器部分应用谱归一化技术,并引入Wasserstein距离和加梯度惩罚的损失函数,增强了模型训练稳定性和收敛速度。实验结果表明,与传统的GAN模型相比,本文模型可以生成更高质量的缺陷甜樱桃图像,两种缺陷甜樱桃图像的FID值(Fréchet inception distance)分别为64.36和59.97。本文模型生成的数据增强后,VGG19和MobileNetV3的甜樱桃分类准确率分别提高16.44个百分点和13.94个百分点。

来源:2024年第10期

《农业机械学报》期刊编辑部

查看农业机械学报杂志2024年第10期

声明

严正声明:本站非期刊官网,非中介代理。

本站仅提供学术规范服务:快速预审、润色编辑服务、中英文查重、降重、去重服务、推荐合适的期刊投稿等学术规范服务。 如需提供学术规范服务请联系在线编辑。

联系我们

  • 地址:北京德胜门外北沙滩1号
  • 电话:86-10-64867367
  • E-mail:njxb@caams.org.cn

咨询工作人员