[发明专利]样本生成方法、目标检测模型训练、目标检测方法及系统在审
申请号: | 202310210125.6 | 申请日: | 2023-03-07 |
公开(公告)号: | CN116152606A | 公开(公告)日: | 2023-05-23 |
发明(设计)人: | 请求不公布姓名;项载蔚;冀春锟 | 申请(专利权)人: | 湖南视比特机器人有限公司 |
主分类号: | G06V10/774 | 分类号: | G06V10/774;G06V10/764;G06V10/20;G06V10/82;G06N3/08;G06N3/0464 |
代理公司: | 长沙湘之星知识产权代理事务所(普通合伙) 43271 | 代理人: | 刘斌;徐仰贵 |
地址: | 410100 湖南省长沙市高新开*** | 国省代码: | 湖南;43 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 样本 生成 方法 目标 检测 模型 训练 系统 | ||
1.一种样本生成方法,其特征在于,包括以下步骤:
步骤1,基于原始样本集训练获得目标检测模型,其中,原始样本集包括多张原始样本图片,目标检测模型的输出为各张原始样本图片上各处的目标检测信息;
步骤2,获取候选样本集,其中,所述候选样本集包括N张待检测的候选样本图片;
步骤3,基于候选样本集和目标检测模型,输出各张候选样本图片上的各处目标信息;
步骤4,基于步骤3的输出信息,从候选样本集中筛选出目标样本图片并确定各目标样本图片上各处的目标位置和类别信息;
步骤5,将目标样本图片及图片上各处的目标位置和类别信息作为训练样本输出。
2.根据权利要求1所述的样本生成方法,其特征在于,图片上各处的目标检测信息包括图片上各处是否有待检测物体以及物体类别概率;
所述步骤3中,针对每张候选样本图片,输出候选样本图片上的各处目标信息过程包括:
步骤301,对候选样本图片做多种可逆的变换处理,获得m张不同的图片P;
步骤302,将所述m张不同的图片P输入目标检测模型,输出各图片P上各处是否有待检测目标以及目标类别概率;
步骤303,基于步骤302的输出结果,计算m张图片P对应的候选样本图片上各处的目标位置和目标类别概率;
步骤304,基于步骤303的计算结果,得到候选样本图片上各处最终的目标位置和目标类别概率信息。
3.根据权利要求2所述的样本生成方法,其特征在于,所述步骤4中包括:
步骤401,以步骤303输出的候选样本图片上各处物体类别概率作为置信度分数;
步骤402,保留全部置信度低于第一预设阈值或高于第二预设阈值的候选样本图片作为目标样本图片输出;将候选样本图片上置信度低于第一预设阈值的位置设为背景,将候选样本图片上置信度高于第二预设阈值的位置设为目标物体并设置物体边界框,丢弃存在置信度位于第一预设阈值和第二预设阈值间的候选样本图片;
步骤403,获取各张图片P对应的候选样本图片上物体边界框对应位置的物体类别概率并取其中的最大值作为物体边界框处的物体类别概率输出。
4.根据权利要求2或3所述的样本生成方法,其特征在于,所述步骤303中,对各图片P上各处物体类别概率求均值,作为候选样本图片上各处物体类别概率。
5.根据权利要求1至3任一项所述的样本生成方法,其特征在于,所述步骤2中,N张待检测区域的候选样本图片,通过对待检测区域连续实时图像采集获得;或者,N张待检测区域的候选样本图片,通过从待检测区域的历史存储图像集中获得。
6.根据权利要求2或3所述的样本生成方法,其特征在于,所述步骤301中,所述处理包括位置翻转处理、图片亮度对比度变化处理、缩小处理或放大处理。
7.一种目标检测模型训练方法,其特征在于,
利用由权利要求1至6任一项所述的样本生成方法生成的训练样本对目标检测模型进行训练,获得更新后的目标检测模型;
或者,利用原始样本集和由权利要求1至6任一项所述的样本生成方法生成的训练样本对目标检测模型进行训练,获得更新后的目标检测模型。
8.一种目标检测模型,其特征在于,所述目标检测模型通过如权利要求7所述的目标检测模型训练方法进行持续自学习更新。
9.一种目标检测方法,其特征在于,利用如权利要求8所述的目标检测模型对待检测图片进行目标检测。
10.一种目标检测系统,其特征在于,包括图像采集单元、模型训练单元和如权利要求8所述的目标检测模型,其中:
图像采集单元:用于采集待检测图片,待检测图片的一部分用于作为候选样本图片用于产生训练样本集以对目标检测模型进行训练更新,待检测图片的另一部分用于借由目标检测模型识别以输出目标检测结果;
模型训练单元:用于基于生成的训练样本集对目标检测模型进行训练以对目标检测模型进行更新。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于湖南视比特机器人有限公司,未经湖南视比特机器人有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202310210125.6/1.html,转载请声明来源钻瓜专利网。
- 上一篇:一种用于医用核酸采样的通风采样亭
- 下一篇:一种双按多换香水扣头手绳