模型权重自更新

关注1人关注
我要报错
  • 基于深度学习的移动端水果识别

    超市水果识别主要依赖人工,计算机视觉成为一种解决方案。然而目前仍面临部分水果识别精度低、终端设备部署困难、误识别图片难处理等挑战。因此,文章基于深度学习对移动端水果识别进行研究,旨在替代人工识别。首先文章构建了包含49种水果的超市水果图像数据集DailyFruit-49。并针对细分类特征相似度高、包装遮挡、形状小量少的水果识别困难,以及低算力设备模型部署问题,筛选了满足部署要求的骨干模型。设计了新的注意力模块RMA,改进了ViT Block以增强模型的细节识别能力和深层语义特征整合能力,最终得到DenseRMA_ViT模型,并基于Focal Loss改进损失函数。并在公开数据集Fruits-262上进行消融实验验证模型改进的有效性。最后结合实际设备,实现水果识别系统,满足实际使用。基于与用户的交互行为对误识别水果图像进行收集,并基于误识别图像实现模型权重自动微调,随使用时间延长,系统收集更多图片,提升模型识别精度与泛化能力,以处理实际应用中误识别水果。