[发明专利]图像数据的处理方法和装置、存储介质及电子设备

说明书

本发明公开了一种图像数据的处理方法和装置、存储介质及电子设备。其中，该方法包括：采用获取待处理的第一图像数据集，其中，第一图像数据集中的图像数据按照第一数据格式进行排列，将第一图像数据集中的数据进行交织重排，得到第二图像数据集，其中，第二图像数据集中的图像数据按照第二数据格式进行排列，交织重排的方式与卷积操作匹配，第二数据格式的维度小于第一数据格式，对第二图像数据集和已获取的第二权重数据集执行卷积操作，得到目标输出结果。本发明解决了相关技术中存在的处理图像数据的效率比较低的技术问题。

技术领域

本发明涉及计算机领域，具体而言，涉及一种图像数据的处理方法和装置、存储介质及电子设备。

背景技术

目前的相关技术中，传统计算模式一般为每一个输出即是一个线程组，利用SIMD（Single Instruction Multiple Data，单指令多数据结构）对图像数据进行处理，例如，在利用SIMD进行卷积运算的过程中，输入维度为[N，C，H，W]=[1，1，5，5]，卷积核维度为[Cout，Cin，kernel_h，kernel_w]=[1，1，3，3]，进行卷积计算并最终生成维度为[N，C，H，W]=[1，1，5，5]的输出，其中，N，C，H，W分别表示批次，通道数，高度，宽度；Cout表示卷积核的输出通道数，Cin表示卷积核的输入通道数，kernel_h表示卷积核的高度，kernel_w表示卷积核的宽度。

现有的技术方案的数据排列通常只针对于[N，C，H，W]的维度，相比传统的数据表达方式对比，对于卷积操作，根据计算机内存排布特点，如果输入空间尺寸较大的话，为了获取信息的完整，需要对输入进行例如补边等操作，进而，由于跨通道获取数据，造成CacheMiss以及额外的数据拷贝开销，导致严重影响计算过程中的设备运行性能。

针对上述的问题，目前尚未提出有效的解决方案。

发明内容

本发明实施例提供了一种图像数据的处理方法和装置、存储介质及电子设备，以至少解决相关技术中存在的处理图像数据的效率比较低的技术问题。

根据本发明实施例的一个方面，提供了一种图像数据的处理方法，包括：获取待处理的第一图像数据集，其中，所述第一图像数据集中的图像数据按照第一数据格式进行排列；将所述第一图像数据集中的数据进行交织重排，得到第二图像数据集，其中，所述第二图像数据集中的图像数据按照第二数据格式进行排列，所述交织重排的方式与卷积操作匹配，所述第二数据格式的维度小于所述第一数据格式；对所述第二图像数据集和已获取的第二权重数据集执行所述卷积操作，得到目标输出结果。

可选地，所述对所述C₂组图像数据和所述第二权重数据集中的N₂×C₂ 组权重数据执行卷积操作，得到所述第三图像数据集中的N₂组图像数据，包括：将所述N₂×C₂ 组权重数据中的每C₂组权重数据分别与所述C₂组图像数据执行加权求和操作，得到所述N₂组图像数据。

根据本发明实施例的另一方面，还提供了一种图像数据的处理装置，包括：获取模块，用于获取待处理的第一图像数据集，其中，所述第一图像数据集中的图像数据按照第一数据格式进行排列；处理模块，用于将所述第一图像数据集中的数据进行交织重排，得到第二图像数据集，其中，所述第二图像数据集中的图像数据按照第二数据格式进行排列，所述交织重排的方式与卷积操作匹配，所述第二数据格式的维度小于所述第一数据格式；执行模块，用于对所述第二图像数据集和已获取的第二权重数据集执行所述卷积操作，得到目标输出结果。