当前位置: 首页 > 专利查询>湖南大学专利>正文

一种多头注意力驱动的厨余垃圾多标签分类方法技术

技术编号:42374547 阅读:15 留言:0更新日期:2024-08-16 14:58
本发明专利技术公开了一种多头注意力驱动的厨余垃圾多标签分类方法,包括:构建厨余垃圾多标签分类数据集,包括厨余垃圾多个不同类别各若干图像,图像标签包括一个或多个类别;构建多头注意力驱动的图卷积轻量化网络模型,包括特征提取模块、多头注意力模块和动态图卷积模块;其中,特征提取模块对输入图像提取特征,然后送入多头注意力模块以加强特征图的类别感知区域,再将其送入动态图卷积模块以自适应捕捉类别感知区域;使用构建的厨余垃圾多标签分类数据集训练所述图卷积轻量化网络模型;最终使用训练得到的分类模型对待预测厨余垃圾图像进行多标签分类。本发明专利技术在降低模型参数量减少带来的性能损失的同时,增强识别能力,提高多标签分类效果。

【技术实现步骤摘要】

本专利技术涉及图像处理领域,具体涉及一种多头注意力驱动的厨余垃圾多标签分类方法


技术介绍

1、近年来,深度学习作为提高垃圾分类效率的手段,许多先进的分类算法被提出,然而目前公开的垃圾数据集大多是基于生活垃圾识别而设计的,对真实场景下厨余垃圾分类缺少相关研究,此外,厨余垃圾图像往往包含多个类别,是计算机视觉领域中典型的多标签图像分类。多标签图像分类作为计算机视觉领域任务之一,主要任务是在于准确预测图像中所包含的全部类别,由于在真实世界中,物体通常是多个同时出现的,这也更加符合人体的认知常识。

2、从厨余垃圾多标签分类角度出发,真实场景对模型的实时性要求较高,由于图像背景复杂,物体类别多样,物体标签之间存在关联性,目前的方法针对该问题的处理若是想获得较高的分类精度往往需要较复杂的网络模型,而基本的深度卷积网络在分类精度上无法保证。因此,研究真实场景下智能化的厨余垃圾多标签高效分类算法具有重要的实际价值和意义。


技术实现思路

1、本专利技术提供一种多头注意力驱动的厨余垃圾多标签分类方法,在降低模型参数本文档来自技高网...

【技术保护点】

1.一种多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,包括:

2.根据权利要求1所述的多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,基于不同季节厨余垃圾的类别分布不同,所述厨余垃圾多标签分类数据集,从一年不同时间段中挑选多种不同类别组合的厨余垃圾图像得到。

3.根据权利要求1所述的多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,所述轻量化的特征提取模块,使用轻量化主干网络ShufflenetV2。

4.根据权利要求1所述的多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,所述多头注意力模块,包括第一全连接层、缩放点积注意力子模块、第二...

【技术特征摘要】

1.一种多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,包括:

2.根据权利要求1所述的多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,基于不同季节厨余垃圾的类别分布不同,所述厨余垃圾多标签分类数据集,从一年不同时间段中挑选多种不同类别组合的厨余垃圾图像得到。

3.根据权利要求1所述的多头注意力驱动的厨余垃圾多标签分类方法,其特征在于,所述轻量化的特征提取模块,使用轻量化主干网络shufflenetv2。

...

【专利技术属性】
技术研发人员:梁桥康李进涛秦海刘铭峰柳力元
申请(专利权)人:湖南大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1