说明:收录25万 73个行业的国家标准 支持批量下载
(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211021535.8 (22)申请日 2022.08.24 (71)申请人 中国联合网络通信集团有限公司 地址 100033 北京市西城区金融大街21号 申请人 中讯邮电咨询设计院有限公司 (72)发明人 竹梦圆 沈涛 王政 宋齐军  张宝昌 张倩 张昀玮 杨泽昆  庞俊彪 李广彬 李尚霖 刘健苗  (74)专利代理 机构 北京中博世 达专利商标代理 有限公司 1 1274 专利代理师 申健 (51)Int.Cl. G06V 30/422(2022.01) G06T 7/70(2017.01) G06N 20/00(2019.01) (54)发明名称 一种置信度阈值的确定方法、 装置及计算机 可读存储介质 (57)摘要 本发明提供一种置信度阈值的确定方法、 装 置及计算机可读存储介质, 涉及深度学习技术领 域, 用于解决置信度阈值的设置未能考虑待检测 对象对查全率和查准率的要求的问题。 该方法包 括: 获取样 本图像, 以及样 本图像中目标类别的P 个真值框的坐标信息; 目标类别为样本图像的待 检测对象类别中的任意一个; 将样 本图像输入至 目标检测模 型, 以目标检测模型对样本图像进行 目标检测, 得到目标类别的M个预测结果; 基于P 个真值框的坐标信息以及M个预测框的坐标信 息, 从M个预测结果中筛选出N个可接受预测结 果; 可接受预测结果为包含的预测框与真值框 之 间的交并比满足要求的预测结果; 基于各个可接 受预测结果所包含的置信度, 确定目标类别的置 信度阈值。 权利要求书2页 说明书9页 附图3页 CN 115359506 A 2022.11.18 CN 115359506 A 1.一种置信度阈值的确定方法, 其特 征在于, 所述方法包括: 获取样本图像, 以及所述样本图像 中目标类别的P个真值框的坐标信 息, 所述目标类别 为所述样本图像的待检测对象类别中的任意 一个, P为 正整数; 将所述样本图像输入至目标检测模型, 以所述目标检测模型对所述样本图像进行目标 检测, 得到所述目标类别的M个预测结果, 一个预测结果包括预测框的坐标信息以及置信 度, M为正整数; 基于所述P个真值框的坐标信息以及M个预测框的坐标信息, 从所述M个预测结果中筛 选出N个可接受预测结果, 所述可接受预测结果为包含的预测框与真值框之间的交并比满 足要求的预测结果, N 为小于或等于 M的正整数; 基于各个可接受预测结果所包 含的置信度, 确定所述目标类别的置信度阈值。 2.根据权利要求1所述的方法, 其特征在于, 在所述真值框的数目小于所述预测框的数 目的情况下, 所述基于所述P 个真值框的坐标信息以及M个预测框的坐标信息, 从所述M个预 测结果中筛 选出N个可接受预测结果, 包括: 对于P个真值框中的各个真值框, 基于所述真值框的坐标信息以及各个预测框的坐标 信息, 分别计算所述真值框与各个预测框的交并比; 从所述真值框与各个预测框的交并比中确定最大交并比; 在所述最大交并比大于或等于交并比阈值 时, 以包含所述最大交并比对应的预测框的 预测结果作为所述可接受预测结果。 3.根据权利要求1所述的方法, 其特征在于, 在所述真值框的数目大于或等于所述预测 框的数目的情况下, 所述基于所述P个真值框的坐标信息以及M个预测框的坐标信息, 从所 述M个预测结果中筛 选出N个可接受预测结果, 包括: 对于M个预测框中的各个预测框, 基于所述预测框的坐标信息与各个真值框的坐标信 息, 分别计算所述预测框与各个真值框的交并比; 从所述预测框与各个真值框的交并比中确定最大交并比; 在所述最大交并比大于或等于交并比阈值 时, 以包含所述预测框的预测结果作为所述 可接受预测结果。 4.根据权利要求1至3任一项所述的方法, 其特征在于, 所述基于各个可接受预测结果 所包含的置信度, 确定所述目标类别的置信度阈值, 包括: 以所有可接受预测结果所包含的置信度中的最小置信度作为所述目标类别的置信度 阈值。 5.一种置信度阈值的确定装置, 其特征在于, 包括获取模块、 目标检测模块、 选择模块 以及确定模块; 所述获取模块, 用于获取样本图像, 以及所述样本图像 中目标类别的P个真值框的坐标 信息, 所述目标类别为所述样本图像的待检测对象类别中的任意 一个, P为 正整数; 所述目标检测模块, 用于将所述样本 图像输入至目标检测模型, 以所述目标检测模型 对所述样本图像进 行目标检测, 得到所述目标类别的M个预测结果, 一个预测结果包括预测 框的坐标信息以及置信度, M为 正整数; 所述选择模块, 用于基于所述P个真值框的坐标信息以及M个预测框的坐标信息, 从所 述M个预测结果中筛选出N个可接受预测结果, 所述可接受预测结果为包含的预测框与真值权 利 要 求 书 1/2 页 2 CN 115359506 A 2框之间的交并比满足要求的预测结果, N 为小于或等于 M的正整数; 所述确定模块, 用于基于各个可接受预测结果所包含的置信度, 确定所述目标类别的 置信度阈值。 6.根据权利要求5所述的装置, 其特 征在于, 所述选择模块, 具体用于: 在所述真值框的数目小于所述预测框的数目的情况下, 对于P个真值框中的各个真值 框, 基于所述真值框的坐标信息以及各个预测框的坐标信息, 分别计算所述真值框与各个 预测框的交并比; 从所述真值框与各个预测框的交并比中确定最大交并比; 在所述最大交并比大于或等于交并比阈值 时, 以包含所述最大交并比对应的预测框的 预测结果作为所述可接受预测结果。 7.根据权利要求5所述的装置, 其特 征在于, 所述选择模块, 具体用于: 在所述真值框的数目大于或等于所述预测框的数目的情况下, 对于M个预测框中的各 个预测框, 基于所述预测框的坐标信息与各个真值框的坐标信息, 分别计算所述预测框与 各个真值框的交并比; 从所述预测框与各个真值框的交并比中确定最大交并比; 在所述最大交并比大于或等于交并比阈值 时, 以包含所述预测框的预测结果作为所述 可接受预测结果。 8.根据权利要求5所述的装置, 其特 征在于, 所述确定模块, 具体用于: 以所有可接受预测结果所包含的置信度中的最小置信度作为所述目标类别的置信度 阈值。 9.一种置信度阈值的确定装置, 其特 征在于, 包括存 储器、 处理器; 所述存储器用于存 储计算机执 行指令, 所述处 理器与所述存 储器通过总线连接; 当所述置信度阈值的确定装置运行时, 所述处理器执行所述存储器存储的所述计算机 执行指令, 以使所述置信度阈值的确定装置执 行如权利要求1 ‑4任一项所述的方法。 10.一种计算机可读存储介质, 其特征在于, 所述计算机可读存储介质包括计算机程序 指令, 当所述计算机程序指令在计算机上运行时, 使得所述计算机执行如权利要求1至4任 一项所述的方法。权 利 要 求 书 2/2 页 3 CN 115359506 A 3

.PDF文档 专利 一种置信度阈值的确定方法、装置及计算机可读存储介质

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