[发明专利]图像处理方法及装置

权利要求书

1.一种图像处理方法，其中，所述图像处理方法包括：

图像获取步骤，获取图像；

特征提取步骤，根据特征提取方法，提取所述图像的图像特征；

图像识别步骤，根据所述图像特征进行图像识别；

其中，所述特征提取方法利用GPU的并行计算，并沿着不同维度对图像特征进行组合，以使不同图像特征共享权重；所述特征提取方法包括：

获取特征图步骤，获取特征图，所述特征图的参数包括特征图批次数量、特征图通道数量、特征图高度以及特征图宽度；

第一尺寸调整步骤，将所述特征图通道数量缩小第一倍数，并将所述特征图批次数量增大所述第一倍数，得到第一张量，其中所述第一倍数为可整除所述特征图通道数量的正整数；

第一卷积步骤，对所述第一张量进行第一卷积操作，得到第二张量；

第二尺寸调整步骤，将所述第二张量的通道数量增大所述第一倍数，并将所述第二张量的批次数量缩小所述第一倍数，得到第三张量；

第三尺寸调整步骤，将所述第三张量的通道数量缩小第二倍数，并将所述第三张量的高度或宽度增大所述第二倍数，得到第四张量，其中所述第二倍数为可整除所述第三张量的通道数量的正整数；

第二卷积步骤，对所述第四张量进行第二卷积操作，得到第五张量；

第四尺寸调整步骤，将所述第五张量的通道数量增大所述第二倍数，并将所述第五张量的高度或宽度缩小所述第二倍数，得到第六张量；

特征提取步骤，基于所述第六张量，得到所述特征图的特征。

2.根据权利要求1所述的图像处理方法，其中，所述第一卷积操作采用m*m卷积核进行卷积，其中m≥3。

3.根据权利要求1所述的图像处理方法，其中，所述第二卷积操作采用点卷积。

4.根据权利要求1所述的图像处理方法，其中，所述第二张量的通道数量为所述第一张量的通道数量的第三倍数，其中所述第三倍数为正整数；

所述第五张量的通道数量相比于所述第四张量的通道数量缩小所述第三倍数。

5.根据权利要求1所述的图像处理方法，其中，所述第一卷积操作共享权重。

6.根据权利要求1所述的图像处理方法，其中，所述第二卷积操作共享权重。

7.一种图像处理装置，其中，所述图像处理装置包括：

图像获取模块，用于获取图像；

特征提取模块，用于提取所述图像的图像特征；

图像识别模块，用于根据所述图像特征进行图像识别；

其中，所述特征提取模块利用GPU的并行计算，并沿着不同维度对图像特征进行组合，以使不同图像特征共享权重；所述特征提取模块包括：

获取特征图模块，用于获取特征图，所述特征图的参数包括特征图批次数量、特征图通道数量、特征图高度以及特征图宽度；

第一尺寸调整模块，用于将所述特征图通道数量缩小第一倍数，并将所述特征图批次数量增大所述第一倍数，得到第一张量，其中所述第一倍数为可整除所述特征图通道数量的正整数；

第一卷积模块，用于对所述第一张量进行第一卷积操作，得到第二张量；

第二尺寸调整模块，用于将所述第二张量的通道数量增大所述第一倍数，并将所述第二张量的批次数量缩小所述第一倍数，得到第三张量；

第三尺寸调整模块，用于将所述第三张量的通道数量缩小第二倍数，并将所述第三张量的高度或宽度增大所述第二倍数，得到第四张量，其中所述第二倍数为可整除所述第三张量的通道数量的正整数；

第二卷积模块，用于对所述第四张量进行第二卷积操作，得到第五张量；

第四尺寸调整模块，用于将所述第五张量的通道数量增大所述第二倍数，并将所述第五张量的高度或宽度缩小所述第二倍数，得到第六张量；

特征提取模块，用于基于所述第六张量，得到所述特征图的特征。

8.一种电子设备，其中，所述电子设备包括：

存储器，用于存储指令；以及

处理器，用于调用所述存储器存储的指令执行如权利要求1至6中任一项所述的图像处理方法。

9.一种计算机可读存储介质，其中存储有指令，所述指令被处理器执行时，执行如权利要求1至6中任一项所述的图像处理方法。