一种结合注意力互斥正则的图像分类方法与流程

本发明涉及图像处理，具体涉及一种结合注意力互斥正则的图像分类方法。

背景技术：

1、图像分类是根据不同类别图像信息中所反映的不同特征，把不同类别的目标区分开来的图像处理方法。目前的图像分类方法，通常采用神经网络模型作为手段，但是在神经网络模型的训练过程中，神经网络模型对图片多个目标区域的关注仍不够理想，例如图3中的第二列模型只关注单个区域等，模型性能有待优化。

技术实现思路

1、为了解决以上问题的一个或多个，提供一种结合注意力互斥正则的图像分类方法。

2、根据本发明的一个方面，提供了一种结合注意力互斥正则的图像分类方法，包括：

3、导入图像；

4、提取图像特征；

5、对图像特征进行分类；

6、所述对图像特征进行分类包括利用图像分类模型的最终损失函数更新模型参数。其有益效果是：结合注意力互斥正则的图像分类方法的训练阶段同时也是构建图像分类模型的过程，而在测试阶段，图像输入该模型后，可以得知所输入的图像类别。

7、在一些实施方式中，所述导入图像包括将用于训练的图像数据集输入图像分类模型中。所导入的训练图像数据集需包括不同类别的训练图像。其有益效果是：让图像分类模型在训练阶段学习不同类别的训练图像模式。

8、在一些实施方式中，所述提取图像特征包括将一张训练图像通过cnn网络提取图像特征，从而得到特征图。由于需根据所提取的图像特征进行区分不同类别的训练图像。其有益效果是：有助于区分不同类别的训练图像。

9、在一些实施方式中，所述提取图像特征包括先将所述特征图转为注意力图，挑选注意力图中指定数目的注意力通道。注意力图对应的是关注训练图像上的哪些区域。其有益效果是：有利于体现对训练图像不同区域的关注。

10、在一些实施方式中，所述提取图像特征还包括：

11、判断注意力通道是否为候选关键通道和限制候选关键注意力通道关注互不重叠的区域。其有益效果是：可以更好地根据图像特征来对图像进行分类。

12、在一些实施方式中，所述判断注意力通道是否为候选关键通道包括：

13、选取一个值作为阈值，若注意力通道的权重大于该阈值，则判断该注意力通道为候选关键通道。注意力通道所对应的权重越大说明注意力通道对应的训练图像区域越重要。其有益效果是：有助于选取候选关键区域。

14、在一些实施方式中，所述限制注意力通道关注互不重叠的区域包括：

15、计算注意力互斥正则损失函数，所述注意力互斥正则损失函数根据以下公式进行计算：

16、

17、其中，lame为注意力互斥正则损失函数，mc1为注意力热区域图的第一注意力通道，mc2为注意力热区域图的第二注意力通道，w为注意力热区域图的宽，h为注意力热区域图的高。不同的注意力通道对应图像不同的区域。其有益效果是：可以关注到不同的区域，有利于整合各候选关键区域的信息。

18、在一些实施方式中，所述对图像特征进行分类还包括：

19、对所述注意力图和特征图执行特征融合操作得到图像的最终特征；

20、对图像的最终特征执行多分类操作，得到输入图像的类别；

21、计算图像分类模型的最终损失函数。其有益效果是：根据所得到的图像特征对图像分类。

22、在一些实施方式中，图像分类模型的最终损失函数包括注意力互斥正则损失函数和交叉熵损失函数。其有益效果是：计算总损失函数可以用于更新图像分类模型的参数。

23、根据本申请的另一个方面，提供一种存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时所述的一种结合注意力互斥正则的图像分类方法的步骤。其有益效果是：运用所述计算机程序对训练图像进行分类。

24、本发明的结合注意力互斥正则的图像分类方法，模型关注到图像目标的不同区域，通过限制模型关注各注意力通道互不重叠的多个关键候选区域，并整合各候选关键区域的信息，提高模型的性能，继而提高图像分类的准确性。

技术特征：

1.一种结合注意力互斥正则的图像分类方法，其特征在于，包括：

2.根据权利要求1所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述导入图像包括将用于训练的图像数据集输入图像分类模型中。

3.根据权利要求1所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述提取图像特征包括将一张训练图像通过cnn网络提取图像特征，从而得到特征图。

4.根据权利要求1所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述提取图像特征包括先将所述特征图转为注意力图，挑选注意力图中指定数目的注意力通道。

5.根据权利要求1所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述提取图像特征还包括：

6.根据权利要求5所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述判断注意力通道是否为候选关键通道包括：

7.根据权利要求5所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述限制注意力通道关注互不重叠的区域包括：

8.根据权利要求4所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述对图像特征进行分类包括：

9.根据权利要求8所述的一种结合注意力互斥正则的图像分类方法，其特征在于，所述图像分类模型的最终损失函数包括注意力互斥正则损失函数和交叉熵损失函数。

10.一种存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时权利要求1至9任一项所述的一种结合注意力互斥正则的图像分类方法的步骤。

技术总结
本发明公开一种结合注意力互斥正则的图像分类方法，涉及图像处理技术领域，包括：导入图像；提取图像特征；提取图像特征包括挑选注意力图中指定数目的若干注意力通道；对图像特征进行分类；对图像特征进行分类包括利用图像分类模型的最终损失函数更新模型参数。优点在于，通过限制模型关注各注意力通道互不重叠的多个关键候选通道，引导模型关注图像目标的不同区域、并整合不同关键区域的信息，提高模型的性能，继而提高图像分类的准确性。

技术研发人员：陆靖桥,宾炜,麦广柱,陶彦百,罗志鹏,陈银
受保护的技术使用者：广东省新黄埔中医药联合创新研究院
技术研发日：
技术公布日：2024/1/13