[发明专利]一种数据处理算法的实现方法、装置、设备和介质

说明书

本公开提供了一种数据处理算法的实现方法、装置、设备和介质，涉及计算机技术领域，尤其涉及计算机视觉技术和芯片技术。具体实现方案为：通过主机获取图像的待处理数据；其中，所述待处理数据包括所述图像中至少两个检测框的检测数据；通过所述主机调用异构设备，根据各检测框的检测数据，执行检测框聚类算子，以得到检测框的聚类结果；其中，所述检测框聚类算子用于对所述检测数据串行执行至少两种算法规则来进行聚类；通过所述主机获取所述异构设备传输反馈的聚类结果，并根据所述聚类结果确定所述图像的数据处理结果。本公开减少了异构设备执行算子的数量和计算规模，提高了数据处理算法的实现效率。

技术领域

本公开涉及计算机技术领域，尤其涉及计算机视觉技术和芯片技术。

背景技术

非极大值抑制(Non-Maximum Suppression，NMS)是一种典型的数据处理算法。该算法顾名思义就是抑制不是极大值的元素，可以理解为局部最大搜索，通常是针对多个检测框进行分值的最大搜索，以便进行检测框的筛选和聚合。

传统NMS算法的实现，使用CPU在主机端计算，其耗时较大；而在基于异构设备的实现中，实现过程较为复杂，传输开销大。

发明内容

本公开提供了一种数据处理算法的实现方法、装置、设备和介质。

根据本公开的一方面，提供了一种数据处理算法的实现方法，该方法包括：

通过主机获取图像的待处理数据；其中，所述待处理数据包括所述图像中至少两个检测框的检测数据；

通过所述主机调用异构设备，根据各检测框的检测数据，执行检测框聚类算子，以得到检测框的聚类结果；其中，所述检测框聚类算子用于对所述检测数据串行执行至少两种算法规则来进行聚类；

通过所述主机获取所述异构设备传输反馈的聚类结果，并根据所述聚类结果确定所述图像的数据处理结果。。

根据本公开的另一方面，提供了一种数据处理算法的实现装置，该装置包括：

待处理数据获取模块，用于通过主机获取图像的待处理数据；其中，所述待处理数据包括所述图像中至少两个检测框的检测数据；

聚类模块，用于通过所述主机调用异构设备，根据各检测框的检测数据，执行检测框聚类算子，以得到检测框的聚类结果；其中，所述检测框聚类算子用于对所述检测数据串行执行至少两种算法规则来进行聚类；

处理结果获取模块，用于通过所述主机获取所述异构设备传输反馈的聚类结果，并根据所述聚类结果确定所述图像的数据处理结果。

根据本公开的另一方面，提供了一种电子设备，包括：

至少两个处理器，所述处理器包括主机和至少一个异构设备；以及

与所述至少两个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少两个处理器执行的指令，所述指令被所述至少两个处理器执行，以使所述至少两个处理器能够执行本公开任一实施例所述的数据处理算法的实现方法。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行本公开任一实施例所述的数据处理算法的实现方法。

根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现本公开任一实施例所述的数据处理算法的实现方法。

本公开实施例可以减少异构设备执行算子的数量和计算规模，提高算法的计算性能。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明