您所在的位置: 成果库 一种基于离散小波注意力模块的图像分类方法及系统

一种基于离散小波注意力模块的图像分类方法及系统

发布时间: 2022-05-24

基本信息

合作方式： 技术转让
成果类型： 发明专利,新技术
行业领域：
新一代信息技术产业,信息传输、软件和信息技术服务业
成果介绍

本发明提出了一种基于离散小波注意力模块的图像分类方法及系统，属于图像分类技术领域，包括构建训练数据集；构建图像分类模型；图像分类模型包括依次连接的特征图提取模块、特征提取模块和特征分类模块；在特征提取模块中包括离散小波注意力模块，离散小波注意力模块基于离散小波变换机制和空间注意力机制，对特征图提取模块输出的特征图进行图像特征的提取，并将提取得到的图像特征输入特征分类模块，得到图像分类结果；利用训练数据集对图像分类模型进行训练，得到训练好的图像分类模型；将待分类的图像尺寸调整为与训练图像相同的尺寸，输入到训练好的图像分类模型中，得到图像分类结果。

成果亮点
团队介绍
成果资料