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

权利要求书

1.一种图像卷积处理方法，其特征在于，包括：

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，

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

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

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

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

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

如果所述第二线程块数量大于或者等于所述第一线程块数量，则确定当前硬件平台的硬件资源，满足使用所述第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求；

如果所述第二线程块数量小于所述第一线程块数量，则确定当前硬件平台的硬件资源，不满足使用所述第一目标预设卷积分块方式进行图像卷积处理时对硬件资源的需求。

3.根据权利要求2所述的方法，其特征在于，

所述计算使用所述第一目标预设卷积分块方式进行图像卷积处理时，需要的线程块的第一线程块数量的步骤，包括：

基于使用所述第一目标预设卷积分块方式进行图像卷积处理时，所述待卷积图像矩阵的一个分块矩阵的尺寸和所述卷积核矩阵的一个分块矩阵的尺寸，确定第一分块乘积矩阵的尺寸；所述第一分块乘积矩阵为使用所述第一目标预设卷积分块方式进行图像卷积处理时，所述待卷积图像矩阵的一个分块矩阵和所述卷积核矩阵的一个分块矩阵的乘积；

对所述待卷积图像矩阵的行数与所述第一分块乘积矩阵的行数的比值进行向上取整操作，得到第一比值；

对所述卷积核矩阵的列数与第一分块乘积矩阵的列数的比值进行向上取整操作，得到第二比值；

将所述第一比值和所述第二比值的乘积，确定为使用所述第一目标预设卷积分块方式进行图像卷积处理时，需要的线程块的第一线程块数量。

4.根据权利要求2或3所述的方法，其特征在于，

所述使用所述第一目标预设卷积分块方式，对所述待卷积图像和卷积核进行图像卷积处理，得到图像卷积结果的步骤，包括：

按照所述第一目标预设卷积分块方式中的划分方法，将所述待卷积图像矩阵和所述卷积核矩阵分别划分为多个第一待卷积图像分块矩阵和多个第一卷积核分块矩阵；

分别计算所述多个第一待卷积图像分块矩阵和所述多个第一卷积核分块矩阵的乘积，得到多个所述第一分块乘积矩阵；

对所述多个第一分块乘积矩阵进行累加运算，得到乘积矩阵；

对所述乘积矩阵进行反向映射处理，得到图像卷积结果；

所述使用所述第二目标预设卷积分块方式对所述待卷积图像和卷积核进行图像卷积处理，得到图像卷积结果的步骤，包括：

按照所述第二目标预设卷积分块方式中的划分方法，将所述待卷积图像矩阵和所述卷积核矩阵分别划分为多个第二待卷积图像分块矩阵和多个第二卷积核分块矩阵；

分别计算所述多个第二待卷积图像分块矩阵和所述多个第二卷积核分块矩阵的乘积，得到多个第二分块乘积矩阵；

对所述多个第二分块乘积矩阵进行累加运算，得到乘积矩阵；

对所述乘积矩阵进行反向映射处理，得到图像卷积结果。