Oct, 2023

高效苹果成熟度和损伤评估:具有 GAN 和注意力机制的轻量级检测模型

TL;DR本研究提出了一种基于轻量级卷积神经网络(CNN)和生成对抗网络(GAN)的方法,用于苹果成熟度和损伤程度检测任务,通过优化模型的深度和宽度以及应用先进的模型压缩技术设计了轻量级 CNN 模型,引入注意机制提高对象检测任务的性能,利用 GAN 生成逼真的苹果图像扩展训练数据集,改善样本不平衡和样本量不足问题,通过在受损苹果上应用目标检测网络进行损伤位置注释,提高了损伤程度检测的准确性,在苹果成熟度和损伤程度检测任务中,该方法的性能优于其他主流模型,具有卓越的性能和高实用价值。