[发明专利]一种最大池化运算的方法及装置

说明书

本申请提供一种最大池化运算的方法及装置，采用芯片中的硬件资源计算一维的最大池化结果及对应的相对索引，由于芯片的硬件只能先读取一个区域内的数据，计算该区域中的最大值并得到相对索引，因此，对图像进行第一方向的连续池化运算，得到一维池化结果和一次相对索引，再计算得到一维池化索引。之后，对一维池化结果进行第二方向的连续池化运算，得到二维池化结果和二次相对索引，再计算得到一维池化索引的索引，最后根据一维池化索引及一维池化索引的索引得到二维池化索引。本实施例的最大池化运算通过芯片中的硬件资源计算，能够同时得到最大池化结果和对应的索引值，不依赖于GPU或CPU等通用控制单元，减少了计算量且控制逻辑易于实现。

技术领域

本申请涉及集成电路技术设计领域，具体而言，涉及一种最大池化运算的方法及装置。

背景技术

最大池化是神经网络中常用的运算操作，他可以实现降低图像的感受野，实现图像的降维，通常会在卷积运算的后面增加一个池化的操作，例如最大池化就是用于计算最大池化的结果，在一些特殊的网络模型中，在得到最大池化的结果的同时，也需要得到最大池化结果在原图中的位置，从而用于将数据恢复成原有图像的尺寸。然而，现有的最大池化运算无法在得到最大池化结果的同时也生成最大池化索引，要想生成最大池化索引通常需要利用CPU或者GPU等通用计算单元的硬件设计方案来实现，这样的硬件设计方案将导致计算量巨大且控制复杂。

发明内容

本申请实施例的目的在于提供一种最大池化运算的方法及装置，用以解决现有的最大池化运算无法在得到最大池化结果的同时也生成最大池化索引的问题。

本申请实施例提供的一种最大池化运算的方法，包括：

对图像进行第一方向的连续池化运算，得到一维池化结果和一次相对索引；

根据一维池化结果和一次相对索引，得到一维池化索引；

对一维池化结果进行第二方向的连续池化运算，得到二维池化结果和二次相对索引；

根据二维池化结果和二次相对索引，得到一维池化索引的索引；其中，一维池化索引的索引用于指示二维池化索引；以及

根据一维池化索引及一维池化索引的索引，得到二维池化索引。

上述技术方案中，可以采用芯片中的硬件资源计算一维的最大池化结果及对应的相对索引，由于芯片的硬件只能先读取一个区域内的数据，计算该区域中的最大值并得到相对索引，因此，对图像进行第一方向的连续池化运算，得到一维池化结果和一次相对索引，再计算得到一维池化索引。之后，对一维池化结果进行第二方向的连续池化运算，得到二维池化结果和二次相对索引，再计算得到一维池化索引的索引，最后根据一维池化索引及一维池化索引的索引得到二维池化索引。因此，本实施例的最大池化运算方法能够同时得到最大池化结果和对应的索引值。

在一些可选的实施方式中，根据一维池化结果和一次相对索引，得到一维池化索引，包括：

对一次相对索引，根据图像第二方向上的尺寸进行放大；

根据图像尺寸及第一方向步进值，得到第一增量；以及

将放大后的一次相对索引与第一增量相加，得到一维池化索引。

在一些可选的实施方式中，根据二维池化结果和二次相对索引，得到一维池化索引的索引，包括：

根据一维池化结果的尺寸及第二方向步进值，得到第二增量；以及

二次相对索引与第二增量相加，得到一维池化索引的索引。

在一些可选的实施方式中，对图像进行第一方向的连续池化运算，得到一维池化结果和一次相对索引，包括：