全网唯一标准王
(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111683471.3 (22)申请日 2021.12.31 (71)申请人 云从科技 集团股份有限公司 地址 511457 广东省广州市南沙区南沙街 金隆路37号5 01房 (72)发明人 潘雪 杨浚琦 吴凡  (74)专利代理 机构 北京瀚仁知识产权代理事务 所(普通合伙) 11482 代理人 宋宝库 郭婷 (51)Int.Cl. G06V 10/774(2022.01) G06V 10/764(2022.01) G06V 10/82(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01)G06K 9/62(2022.01) (54)发明名称 图片分析模 型调整方法、 装置以及计算机可 读存储介质 (57)摘要 本发明涉及计算机处理技术领域, 具体提供 一种图片分析模 型调整方法、 装置以及计算机可 读存储介质, 旨在解决快速调整模 型以降低误检 情况的技术问题。 为此目的, 本发明的方法包括: 将训练图片输入模型, 其携带指示第一检测框的 类别的标注数据, 模型检测头包括第一分支和第 二分支, 计算第二检测框的置信度和类别; 在置 信度高于 预设水平时, 判断第二检测框与第一检 测框类别是否相同, 不相同时为第二检测框设置 伪标签, 记录其类别为未知类别; 对第二分支进 行调整; 调整完毕后, 禁止模型输出检测框类别 为未知类别的结果。 本发明只对输出检测框类别 的分支进行调整, 有利于提升模型调整的效率, 禁止模型输出未知类型的检测结果, 改善了模型 的识别能力。 权利要求书2页 说明书10页 附图5页 CN 114359669 A 2022.04.15 CN 114359669 A 1.一种图片分析模型调整方法, 其特 征在于, 所述方法包括: 将训练图片输入所述模型, 所述训练图片携带指示了所述训练图片中目标对象的第 一 检测框的类别的标注数据, 所述模型 的检测头包括第一分支和第二分支, 分别计算用于指 示所述目标对象的第二检测框的置信度和类别; 在所述第二检测框的置信度高于预设水平时, 判断所述第 二检测框的类别与 所述第一 检测框的类别是否相同, 在不相同时为所述第二检测框设置伪标签, 并在所述伪标签中记 录所述第二检测框的类别为未知类别; 将所述第二检测框及所述伪标签作为所述第二分支 的训练数据, 输入所述第二分支, 根据输出 结果对所述第二分支进行调整; 在所述模型调整完毕后, 禁止所述模型输出检测框类别为所述未知类别的结果。 2.根据权利要求1所述的图片分析模型调整方法, 其特征在于, “根据输出结果对所述 第二分支进行调整 ”的步骤, 包括: 根据预设的损 失函数, 计算所述输出结果的损 失值, 根据所述损 失值对所述第二分支 的参数进行调整。 3.根据权利要求2所述的图片分析模型调整方法, 其特征在于, 在 “在所述模型调整完 毕后, 禁止所述模型输出检测框类别为所述未知类别的结果 ”的步骤之前, 还 包括: 在检测到所述损失值小于预设阈值后, 确定所述模型调整完毕。 4.根据权利要求1所述的图片分析模型调整方法, 其特征在于, 在 “将训练图片输入所 述模型”的步骤之前, 还 包括: 根据所述模型输出的历史错 误结果中的检测框类别, 获取 所述训练图片。 5.根据权利要求1所述的图片分析模型调整方法, 其特征在于, 在 “将所述第二检测框 及所述伪标签作为所述第二分支的训练数据 ”的步骤之前, 还 包括: 如果所述第 二检测框与其他检测框位于同一连通域, 则根据所述其他检测框的类别更 新所述第二检测框的类别。 6.根据权利要求1所述的图片分析模型调 整方法, 其特征在于, 所述标注数据中还指示 了所述第一检测框的位置, 所述检测头还 包括第三分支, 用于计算所述第二检测框的位置 。 7.根据权利要求1所述的图片分析模型调 整方法, 其特征在于, 所述模型还包括特征提 取层, 所述特征提取层包括主干网络和多尺度特征融合网络, 所述主管网络用于从所述训 练图片中提取多尺度特征, 所述多尺度特征融合网络用于将所述训练图片的多尺度特征融 合为用于输入所述检测头的所述训练图片的特 征。 8.一种图片分析模型调整装置, 其特 征在于, 所述装置包括: 图片输入模块, 将训练图片输入所述模型, 所述训练图片携带指示了所述训练图片中 目标对象的第一检测框的类别的标注数据, 所述模型 的检测头包括第一分支和第二分支, 分别计算用于指示所述目标对象的第二检测框的置信度和类别; 类别设置模块, 在所述第二检测框的置信度高于预设水平时, 判断所述第二检测框的 类别与所述第一检测框的类别是否相同, 在不相同时为所述第二检测框设置伪标签, 并在 所述伪标签中记录所述第二检测框的类别为未知类别; 分支调整模块, 将所述第二检测框及所述伪标签作为所述第二分支 的训练数据, 输入 所述第二分支, 根据输出 结果对所述第二分支进行调整;权 利 要 求 书 1/2 页 2 CN 114359669 A 2输出控制模块, 在所述模型调整完毕后, 禁止所述模型输出检测框类别为所述未知类 别的结果。 9.一种控制装置, 包括处理器和存储装置, 所述存储装置适于存储多条程序代码, 其特 征在于, 所述程序 代码适于由所述处理器加载并运行以执行权利要求 1至7中任一项 所述的 图片分析模型调整方法。 10.一种计算机可读存储介质, 其中存储有多条程序代码, 其特征在于, 所述程序代码 适于由处 理器加载并运行以执 行权利要求1至7中任一项所述的图片分析模型调整方法。权 利 要 求 书 2/2 页 3 CN 114359669 A 3

.PDF文档 专利 图片分析模型调整方法、装置以及计算机可读存储介质

文档预览
中文文档 18 页 50 下载 1000 浏览 0 评论 309 收藏 3.0分
温馨提示:本文档共18页,可预览 3 页,如浏览全部内容或当前文档出现乱码,可开通会员下载原始文档
专利 图片分析模型调整方法、装置以及计算机可读存储介质 第 1 页 专利 图片分析模型调整方法、装置以及计算机可读存储介质 第 2 页 专利 图片分析模型调整方法、装置以及计算机可读存储介质 第 3 页
下载文档到电脑,方便使用
本文档由 人生无常 于 2024-03-18 22:44:44上传分享
友情链接
站内资源均来自网友分享或网络收集整理,若无意中侵犯到您的权利,敬请联系我们微信(点击查看客服),我们将及时删除相关资源。