[发明专利]基于算子融合的图像识别方法、装置、设备及存储介质

权利要求书

1.一种基于算子融合的图像识别方法，其特征在于，包括如下步骤：

获取用户输入的模型参数，并根据所述模型参数，建立视觉模型，并对所述视觉模型进行解析，获取算子连接图；

根据预设的融合规则和所述算子连接图，获取全部可能的算子融合策略；

根据预设的优化目标函数，对所述算子融合策略进行筛选，其中，筛选后的算子融合策略指示当前算子融合策略的计算复杂度最低；

根据所述筛选后的算子融合策略执行算子融合操作，获取融合后的视觉模型；

获取待识别图像，并将所述待识别图像输入所述融合后的视觉模型，通过所述融合后的视觉模型，输出图像识别结果。

2.根据权利要求1所述的一种基于算子融合的图像识别方法，其特征在于，所述获取全部可能的算子融合策略，还包括：

根据所述算子融合规则，获取全部可能融合的算子连接组；

对所述视觉模型中的全部算子进行遍历，若所述算子存在于所述可能融合的算子连接组，则进行融合选择判断，若选择根据所述算子连接组进行该算子的融合，则记录融合后的算子，若选择不融合该算子，则记录该算子；

当所述全部算子遍历结束后，生成融合算子连接图，其中，所述融合算子连接图包括全部可能的算子融合策略。

3.根据权利要求2所述的一种基于算子融合的图像识别方法，其特征在于，所述对所述算子融合策略进行筛选，还包括：

根据所述融合算子连接图，通过所述优化目标函数公式，获取所述融合算子连接图的计算复杂度，其中，所述优化目标函数公式为：

其中，op⁽ⁱ⁾ _fused 为融合后的算子，op⁽ⁱ⁾_org为无法融合的算子，n为所有算子的数量，TC(.)为时间复杂度，SC(.)为空间复杂度；

获取计算复杂度最小的融合算子连接图，并确认该融合算子连接图所对应的算子融合策略为筛选后的算子融合策略。

4.根据权利要求1所述的一种基于算子融合的图像识别方法，其特征在于，获取融合后的视觉模型后，还包括：

根据预设的视觉任务与评价指标的对应关系，获取所述视觉模型的评价指标，其中所述视觉任务包括目标分类任务、目标检测任务、目标分割任务以及目标跟踪任务，所述评价指标包括精度评价和速度评价；

根据所述评价指标，对所述融合后的视觉模型进行评价，并根据评价后的评价结果与预设目标评价结果之间的对比，通过模型微调方法，对所述融合后的视觉模型参数进行优化。

5.一种基于算子融合的图像识别装置，其特征在于，包括：

算子连接图获取模块，用于获取用户输入的模型参数，并根据所述模型参数，建立视觉模型，并对所述视觉模型进行解析，获取算子连接图；

算子融合策略获取模块，用于根据预设的融合规则和所述算子连接图，获取全部可能的算子融合策略；

融合策略筛选模块，用于根据预设的优化目标函数，对所述算子融合策略进行筛选，其中，筛选后的算子融合策略指示当前算子融合策略的计算复杂度最低；

算子融合模块，用于根据所述筛选后的算子融合策略执行算子融合操作，获取融合后的视觉模型；

图像识别模块，用于获取待识别图像，并将所述待识别图像输入所述融合后的视觉模型，通过所述融合后的视觉模型，输出图像识别结果。

6.根据权利要求5所述的一种基于算子融合的图像识别装置，其特征在于，所述算子融合策略获取模块，还包括：

算子链接组获取单元，用于根据所述算子融合规则，获取全部可能融合的算子连接组；

融合选择判断单元，用于对所述视觉模型中的全部算子进行遍历，若所述算子存在于所述可能融合的算子连接组，则进行融合选择判断，若选择根据所述算子连接组进行该算子的融合，则记录融合后的算子，若选择不融合该算子，则记录该算子；

融合算子连接图生成单元，用于当所述全部算子遍历结束后，生成融合算子连接图，其中，所述融合算子连接图包括全部可能的算子融合选择。