基于算子融合的图像识别方法、装置、设备及存储介质
摘要:
本发明提供了一种基于算子融合的图像识别方法、装置、电子设备及存储介质,通过获取用户输入的模型参数,建立视觉模型,并对视觉模型进行解析,获取算子连接图;根据预设的融合规则和算子连接图,获取全部可能的算子融合策略;再根据预设的优化目标函数,对算子融合策略进行筛选;根据筛选后的算子融合策略执行算子融合操作,获取融合后的视觉模型;再获取待识别图像,并将待识别图像输入所述融合后的视觉模型,通过融合后的视觉模型,输出图像识别结果。本发明通过对算子融合策略的计算复杂度进行筛选,以快速且高效的方法确定全局最优的算子融合策略,避免了陷入局部最优解的同时,实现了视觉模型计算效率的提升。
0/0