图像识别方法及装置与流程

文档序号:33809241发布日期:2023-04-19 13:12阅读:来源:国知局

技术特征:

1.一种图像识别方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述根据预设的yolov7-e6e识别模型组合训练确定所述yolov7-x识别模型的步骤包括:

3.根据权利要求2所述的方法,其特征在于,所述组合损失数据是所述yolov7-x识别模型和所述yolov7-e6e识别模型最后一层的输出,基于预设的加权参数进行加权组合后的组合识别结果,与所述基准识别结果的差值。

4.根据权利要求2所述的方法,其特征在于,所述分别将所述训练图像集导入未训练的yolov7-x识别模型,以及所述yolov7-x识别模型和所述yolov7-e6e识别模型构成的组合模型进行迭代训练,直至所述组合模型的组合损失数据符合预设要求,得到所述训练好的所述yolov7-x识别模型,包括:

5.根据权利要求4所述的方法,其特征在于,所述基于所述更新率矩阵对所述yolov7-x识别模型进行优化,包括:

6.根据权利要求4所述的方法,其特征在于,所述若表示优化成功,将优化后的yolov7-x识别模型更新为当次的yolov7-x识别模型之后,包括:

7.根据权利要求2所述的方法,其特征在于,所述组合模型组合损失数据符合预设要求是指在最近的第二预设数量次训练中,所述组合损失数据变化的方差小于预设的第四阈值。

8.一种图像识别装置,其特征在于,包括:

9.一种电子设备,包括存储器、处理器以及存储在所述存储器中并且可以在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至7任一项所述方法的步骤。

10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7任一项所述方法的步骤。


技术总结
本公开涉及图像识别技术领域,提供了图像识别方法及装置。该方法包括:获取包含待识别目标的原始图像;将所述原始图像从RGB色域空间转换为LAB色域空间,得到LAB图像;将所述LAB图像导入训练好的yolov7‑x识别模型进行识别,得到针对所述待识别目标的识别结果,其中,所述yolov7‑x识别模型是根据预设的yolov7‑E6E识别模型组合训练确定的,所述yolov7‑E6E识别模型的模型复杂度高于所述yolov7‑x识别模型。本公开实施例可以在减小识别模型的基础上大大提高识别精度,使得可以在满足精度的基础上,在手持式设备、便携式设备等小型设备上进行高效的运行。

技术研发人员:张剑,穆振华,辛浩冬,罗建涛
受保护的技术使用者:国能包神铁路集团有限责任公司
技术研发日:
技术公布日:2024/1/13
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1