[发明专利]一种图像卷积处理方法和电子设备

说明书

本申请实施例提供了一种图像卷积处理方法和电子设备，获取待卷积图像和卷积核；判断当前硬件平台的硬件资源，是否满足使用第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求；若满足，使用第一目标预设卷积分块方式，对待卷积图像和卷积核进行图像卷积处理；若不满足，根据预先设定的卷积核的个数与卷积分块方式的对应关系，从除第一目标预设卷积分块方式之外的其他预设卷积分块方式中，选择与卷积核的个数对应的第二目标预设卷积分块方式，使用第二目标预设卷积分块方式对待卷积图像和卷积核进行图像卷积处理。本申请实施例中，在多种卷积分块方式中选择目标卷积分块方式，提高了图像卷积处理的效率。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像卷积处理方法和电子设备。

背景技术

在图像处理技术领域，经常需要对图像中的数据进行卷积处理，从而达到特征提取的目的。卷积处理的过程为：分别将待卷积图像和卷积核映射为两个矩阵：待卷积图像矩阵和卷积核矩阵，从而将待卷积图像与卷积核之间的卷积运算，转化为上述待卷积图像矩阵和卷积核矩阵的乘法运算，并对矩阵乘法运算得到的乘积矩阵进行反向映射，从而得到图像卷积结果。

目前，一般采用分块卷积的方法获得卷积结果，具体的：按照预先设定的划分方法，将待卷积图像矩阵和卷积核矩阵分别划分为多个待卷积图像分块矩阵和多个卷积核分块矩阵，然后分别计算各待卷积图像分块矩阵和各卷积核分块矩阵的乘积，得到多个分块乘积矩阵，并对上述多个分块乘积矩阵进行累加运算，得到乘积矩阵，再对乘积矩阵进行反向映射，从而得到图像卷积结果。

上述分块卷积方法中，对于任意尺寸的待卷积图像或者卷积核，分块方法是固定的，即：对于任意尺寸的待卷积图像或者卷积核，分块得到的待卷积图像分块矩阵的尺寸均是相同的，卷积核分块矩阵的尺寸均是相同的，相应的，分块乘积矩阵的尺寸也是相同的，而在实际处理过程中，针对部分尺寸的待卷积图像或者卷积核，采用上述分块卷积方法，处理效率较低。

发明内容

本申请实施例的目的在于提供一种图像卷积处理方法、装置、电子设备及存储介质，以提高图像卷积处理的效率。具体技术方案如下：

第一方面，本申请实施例提供了一种图像卷积处理方法，包括：

获取待卷积图像和卷积核；

判断当前硬件平台的硬件资源，是否满足使用第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求；其中，所述第一目标预设卷积分块方式为多种预设卷积分块方式中，使得分块尺寸最小的卷积分块方式；

若满足，则使用所述第一目标预设卷积分块方式，对所述待卷积图像和卷积核进行图像卷积处理，得到图像卷积结果；

若不满足，则根据预先设定的卷积核的个数与卷积分块方式的对应关系，从除所述第一目标预设卷积分块方式之外的其他预设卷积分块方式中，选择与所述卷积核的个数对应的第二目标预设卷积分块方式，使用所述第二目标预设卷积分块方式对所述待卷积图像和卷积核进行图像卷积处理，得到图像卷积结果。

进一步的，在所述判断当前硬件平台的硬件资源，是否满足使用第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求的步骤之前，还包括：

将所述待卷积图像和卷积核分别映射为待卷积图像矩阵和卷积核矩阵；

所述判断当前硬件平台的硬件资源，是否满足使用第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求的步骤，包括：

计算使用所述第一目标预设卷积分块方式进行图像卷积处理时，需要的线程块的第一线程块数量；其中，一个线程块用于对所述待卷积图像矩阵中的一个分块矩阵和所述卷积核矩阵中的一个分块矩阵进行矩阵乘法运算；

计算使用所述第一目标预设卷积分块方式进行图像卷积处理时，基于当前硬件平台的硬件资源所能够运行的线程块的第二线程块数量；