[发明专利]一种图像卷积方法、装置、设备及存储介质

权利要求书

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

对卷积核进行分割处理，得到尺寸适配一维或二维Winograd卷积算子的多个卷积核划块；

对输入特征图进行分割处理，得到各卷积核划块分别对应的输入划块；

将各输入划块及对应的卷积核划块分别输入适配的一维或二维Winograd卷积算子，通过卷积运算得到对应的输出划块；

将各输出划块进行累加，得到输入特征图的卷积输出特征图。

2.如权利要求1所述的图像卷积方法，其特征在于，卷积核划块的尺寸包括2×2、1×2、1×1。

3.如权利要求1所述的图像卷积方法，其特征在于，对输入特征图进行分割处理，得到各卷积核划块分别对应的输入划块，包括：在卷积核滑动过程中，分别提取各卷积核划块对应的输入特征点，组成各卷积核划块对应的输入划块。

4.如权利要求1所述的图像卷积方法，其特征在于，将各输入划块及对应的卷积核划块分别输入适配的一维或二维Winograd卷积算子，通过卷积运算得到各输入划块对应的输出划块，包括： 一维Winograd卷积算子对一维输入划块及对应的卷积核划块进行一维输入转置、 一维点乘、一维输出转置，得到对应的输出划块。

5.如权利要求1所述的图像卷积方法，其特征在于，将各输入划块及对应的卷积核划块分别输入适配的一维或二维Winograd卷积算子，通过卷积运算得到各输入划块对应的输出划块，包括： 二维Winograd卷积算子对二维输入划块及对应的卷积核划块进行二维输入转置、二维点乘、二维输出转置，得到对应的输出划块。

6.一种图像卷积装置，其特征在于，包括：

卷积核分割模块，对卷积核进行分割处理，得到尺寸适配一维或二维Winograd卷积算子的多个卷积核划块；

输入特征图分割模块，对输入特征图进行分割处理，得到各卷积核划块分别对应的输入划块；

Winograd卷积算子模块，对各输入划块及对应的的卷积核划块进行卷积运算，得到对应的输出划块；

累加模块，将各输出划块进行累加，得到输入特征图的卷积输出特征图。

7.如权利要求6所述的图像卷积装置，其特征在于，Winograd卷积算子模块包括一维Winograd卷积算子模块，一维Winograd卷积算子模块包括：

一维输入转置单元，对一维输入划块及对应的卷积核划块进行转置；

一维点乘单元，对转置后的一维输入划块及对应的卷积核划块进行一维点乘计算；

一维输出转置单元，对一维点乘单元的计算结果进行转置得到对应的输出划块。

8.如权利要求6所述的图像卷积装置，其特征在于，Winograd卷积算子模块包括二维Winograd卷积算子模块，二维Winograd卷积算子模块包括：

二维输入转置单元，对二维输入划块及对应的卷积核划块进行转置；

二维点乘单元，对转置后的二维输入划块及对应的卷积核划块进行二维点乘计算；

二维输出转置单元，对二维点乘单元的计算结果进行转置得到对应的输出划块。

9.一种电子设备，其特征在于包括存储器和处理器，所述存储器上存储有可在所述处理器上运行的程序，所述程序在被所述处理器执行时实现权利要求1-5任一项所述的图像卷积方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述存储介质存储有至少一个程序，所述至少一个程序可被至少一个处理器执行，所述至少一个程序在被所述至少一个处理器执行时实现权利要求1-5任一项所述的图像卷积方法的步骤。