图片分析模型调整方法、装置以及计算机可读存储介质制造方法及图纸

技术编号:33131875 阅读:19 留言:0更新日期:2022-04-17 00:49
本发明专利技术涉及计算机处理技术领域,具体提供一种图片分析模型调整方法、装置以及计算机可读存储介质,旨在解决快速调整模型以降低误检情况的技术问题。为此目的,本发明专利技术的方法包括:将训练图片输入模型,其携带指示第一检测框的类别的标注数据,模型检测头包括第一分支和第二分支,计算第二检测框的置信度和类别;在置信度高于预设水平时,判断第二检测框与第一检测框类别是否相同,不相同时为第二检测框设置伪标签,记录其类别为未知类别;对第二分支进行调整;调整完毕后,禁止模型输出检测框类别为未知类别的结果。本发明专利技术只对输出检测框类别的分支进行调整,有利于提升模型调整的效率,禁止模型输出未知类型的检测结果,改善了模型的识别能力。的识别能力。的识别能力。

【技术实现步骤摘要】
图片分析模型调整方法、装置以及计算机可读存储介质


[0001]本专利技术涉及计算机
,具体提供一种图片分析模型调整方法、装置以及计算机可读存储介质。

技术介绍

[0002]随着人工智能领域的发展,计算机视觉技术在生活中得到广泛应用,人们所熟知的刷脸支付、智能监控、自动驾驶等等,背后都有一套基于计算机视觉的系统提供支持。这些系统的第一步工作大部分都是目标检测任务,找出图像中感兴趣的目标(物体),确定他们的类别和位置,然后再交由系统的后续识别、跟踪等模块进一步处理。这种检测模型通常只针对特定的检测类别,如人脸、车辆、行人、商品等,模型可以部署在开放场景下的任意环境中。
[0003]但是基于特定任务的检测模型学习都会面临一个通用的问题,准备阶段训练数据场景单一,即使在训练集表现较好的模型,当部署在开放场景时也会存在很多意想不到的误检。因此模型在实际部署过程中,场景多样性和复杂性急剧增大,误检风险明显升高。而误检的存在又会极大地影响系统的后续识别结果。因此,如何在有限的训练集上,迅速迭代模型,减少模型在开放场景的误检是一个非常有意义且十分本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图片分析模型调整方法,其特征在于,所述方法包括:将训练图片输入所述模型,所述训练图片携带指示了所述训练图片中目标对象的第一检测框的类别的标注数据,所述模型的检测头包括第一分支和第二分支,分别计算用于指示所述目标对象的第二检测框的置信度和类别;在所述第二检测框的置信度高于预设水平时,判断所述第二检测框的类别与所述第一检测框的类别是否相同,在不相同时为所述第二检测框设置伪标签,并在所述伪标签中记录所述第二检测框的类别为未知类别;将所述第二检测框及所述伪标签作为所述第二分支的训练数据,输入所述第二分支,根据输出结果对所述第二分支进行调整;在所述模型调整完毕后,禁止所述模型输出检测框类别为所述未知类别的结果。2.根据权利要求1所述的图片分析模型调整方法,其特征在于,“根据输出结果对所述第二分支进行调整”的步骤,包括:根据预设的损失函数,计算所述输出结果的损失值,根据所述损失值对所述第二分支的参数进行调整。3.根据权利要求2所述的图片分析模型调整方法,其特征在于,在“在所述模型调整完毕后,禁止所述模型输出检测框类别为所述未知类别的结果”的步骤之前,还包括:在检测到所述损失值小于预设阈值后,确定所述模型调整完毕。4.根据权利要求1所述的图片分析模型调整方法,其特征在于,在“将训练图片输入所述模型”的步骤之前,还包括:根据所述模型输出的历史错误结果中的检测框类别,获取所述训练图片。5.根据权利要求1所述的图片分析模型调整方法,其特征在于,在“将所述第二检测框及所述伪标签作为所述第二分支的训练数据”的步骤之前,还包括:如果所述第二检测框与其他检测框位于同一连通域,则根据所述其他检测框的类别更新所述第二检测框的类别。6.根据权利要求1所...

【专利技术属性】
技术研发人员:潘雪杨浚琦吴凡
申请(专利权)人:云从科技集团股份有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1