[发明专利]一种池化处理方法及装置、存储介质

说明书

本申请实施例提供了一种池化处理方法及装置、存储介质，包括：在待池化图片的目标通道中获取多个内存地址，多个内存地址的个数与池化窗口的边长相同；对多个内存地址依次进行交叉存取，确定出多个内存地址对应的多组内存地址，并从多组内存地址中获取多组数据，多个内存地址中的一个内存地址对应一组内存地址，多组数据中的一组数据的个数由存储单元的位宽确定；将多组数据划分为多个通道对应的多组待池化数据，多个通道包括目标通道，多个通道中的一个通道对应一组待池化数据；按照池化窗口，对多个通道中的任一通道对应的一组待池化数据进行相应的池化处理。

技术领域

本申请涉及图像处理领域，尤其涉及一种池化处理方法及装置、存储介质。

背景技术

近几年来，作为具有自组织、自学习和联想功能的神经网络已成功地应用于图像处理的许多方面，如图像压缩、图像分割、边缘检测、图像增强、图像识别等，而池化是神经网络中一个不可或缺的操作，由于图像具有“静态型”的属性，即在一个图像区域的特征极有可能在另一个区域同样适用，故可以通过对不同位置的特征进行聚合统计的池化方式来描述一幅图像，该池化方式可以包括计算一个图像区域中某个特征的平均值或者最大值，采用这种池化方式可以保留有用信息而去除冗余信息，有利于后续提取有效信息。

目前神经网络中内存布局主要有NCHW和NHWC两种。对于NCHW格式，其内存布局示例性的为(RRRR GGGG BBBB)，在对NCHW进行池化操作时，由于其内存布局连续，多采用“滑窗法”，而NHWC格式其内存布局示例性的为(RGB RGB RGB RGB)，可以看出其多通道内存不连续，无法采用“滑窗法”，需要对每个待比较的值先计算其内存地址，故，对NHWC进行池化操作时需要间隔获取内存，进而导致内存的高速缓冲存储器cache的命中率降低，池化速度降低。

发明内容

本申请实施例提供一种池化处理方法及装置、存储介质，能够提高池化速度。

本申请的技术方案是这样实现的：

本申请实施例提供一种池化处理方法，所述方法包括：

在待池化图片的目标通道中获取多个内存地址，所述多个内存地址的个数与池化窗口的边长相同，其中，所述待池化图片按照NHWC布局类型进行内存布局，N为图片数量，C为通道数量，H为图片高度，W为图片宽度；

对所述多个内存地址依次进行交叉存取，确定出所述多个内存地址对应的多组内存地址，并从所述多组内存地址中获取多组数据，所述多个内存地址中的一个内存地址对应一组内存地址，所述多组数据中的一组数据的个数由存储单元的位宽确定；

将所述多组数据划分为多个通道对应的多组待池化数据，所述多个通道包括所述目标通道，所述多个通道中的一个通道对应一组待池化数据；

按照所述池化窗口，对所述多个通道中的任一通道对应的一组待池化数据进行相应的池化处理。

在上述方法中，所述对所述多个内存地址依次进行交叉存取，确定出所述多个内存地址对应的多组内存地址，包括：

根据所述存储单元的位宽和内存地址中数据的位宽，依次确定出所述多个内存地址中第一内存地址对应的一组内存地址的第一数量；

使用单指令多数据流指令NEON指令对第一内存地址进行交叉存取，得到一组第一数量的内存地址；

根据所述多个内存地址对应的多个一组第一数量的数据，确定出所述多组内存地址。

在上述方法中，所述将所述多组数据划分为多个通道对应的多组待池化数据之后，所述方法还包括：

将所述多个通道对应的所述多组待池化数据分别存储至多个存储单元中，所述多个存储单元中的一个存储单元存储一个通道对应的一组待池化数据。