[发明专利]用于目标检测任务的加速框架生成方法、装置及电子设备

说明书

本申请公开了一种用于目标检测任务的加速框架生成方法、装置及电子设备，涉及人工智能技术领域，尤其涉及深度学习、计算机视觉领域。具体实现方案为：获取目标检测任务的多个处理流程；根据多个处理流程，确定目标检测任务所使用的目标检测算法流程所占用的N个不同硬件处理单元，其中，N为正整数；根据N个不同硬件处理单元，创建N条流水线；其中，每个硬件处理单元对应一条流水线；根据多个处理流程和N条流水线，生成目标检测任务的加速框架。本申请可以提高硬件处理单元的利用率和算法执行效率，实现对于目标检测算法性能的充分发挥。

技术领域

本申请涉及人工智能技术领域，尤其涉及深度学习、计算机视觉领域，特别的涉及一种用于目标检测任务的加速框架生成方法、装置、电子设备和存储介质。

背景技术

目标检测是将目标的分割和识别合二为一的处理方法，其广泛应用于机器人导航、智能视频监控、工业检测、航空航天等诸多领域。

目前，在AI(Artificial Intelligence人工智能)芯片上有CPU(CentralProcessing Unit中央处理器)、NPU(Neural-network Processing Unit网络处理器)或GPU(Graphic Processing Unit图像处理器)、图像处理等多个计算单元。相关技术中，目标检测在读取摄像头进行实时检测时，其中就可能涉及CPU、NPU、图像处理等多个计算单元的计算。如何提高目标检测的算法性能已经成为目标检测过程中亟待解决的技术问题。

发明内容

本申请提供了一种用于目标检测任务的加速框架生成方法、装置及电子设备。

根据本申请的第一方面，提供了一种用于目标检测任务的加速框架生成方法，包括：

获取目标检测任务的多个处理流程；

根据所述多个处理流程，确定所述目标检测任务所使用的目标检测算法流程所占用的N个不同硬件处理单元，其中，所述N为正整数；

根据所述N个不同硬件处理单元，创建N条流水线；其中，每个所述硬件处理单元对应一条流水线；

根据所述多个处理流程和所述N条流水线，生成所述目标检测任务的加速框架。

根据本申请的第二方面，提供了一种用于目标检测任务的加速框架生成装置，包括：

获取模块，用于获取目标检测任务的多个处理流程；

确定模块，用于根据所述多个处理流程，确定所述目标检测任务所使用的目标检测算法流程所占用的N个不同硬件处理单元，其中，所述N为正整数；

创建模块，用于根据所述N个不同硬件处理单元，创建N条流水线；其中，每个所述硬件处理单元对应一条流水线；

生成模块，用于根据所述多个处理流程和所述N条流水线，生成所述目标检测任务的加速框架。

根据本申请的第三方面，提供了一种电子设备，包括：

至少一个处理器；以及

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

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行前述第一方面所述的方法。

根据本申请的第四方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，所述计算机指令用于使所述计算机执行前述第一方面所述的方法。

根据本申请的第五方面，提供了一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现根据前述第一方面所述的方法。