一种基于SE-ResNet模型的茶叶品种分类方法

    公开(公告)号:CN114140652A

    公开(公告)日:2022-03-04

    申请号:CN202111569552.0

    申请日:2021-12-21

    Abstract: 本发明公开了一种基于SE‑ResNet模型的茶叶品种分类方法,包括采集茶叶原始图片,根据SE‑ResNet模型设计要求等比例缩放图片数据,依次采用翻转变换、旋转变换、噪声扰动和对比度变换方式扩大图片数据的泛化性;再确定网络层数后,选择超参数、激活函数和损失函数构建茶叶品种分类的SE‑ResNet模型,采用训练集和验证集数据训练上述SE‑ResNet模型,保存对应的训练权重,根据训练精度和损失函数图调优上述模型;将训练权重重新加载到上述调优后的模型中,最后用测试集数据验证该模型对茶叶品种分类性能。本发明可以快速识别出不同的茶叶品种,提高茶叶品种分类的效率,减少时间和人力成本。

Patent Agency Ranking