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