[发明专利]一种数据处理方法、装置及设备在审
| 申请号: | 202110956614.7 | 申请日: | 2021-08-19 |
| 公开(公告)号: | CN113705648A | 公开(公告)日: | 2021-11-26 |
| 发明(设计)人: | 王金 | 申请(专利权)人: | 杭州海康威视数字技术股份有限公司 |
| 主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N3/04 |
| 代理公司: | 北京博思佳知识产权代理有限公司 11415 | 代理人: | 杨春香 |
| 地址: | 310051 浙*** | 国省代码: | 浙江;33 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 一种 数据处理 方法 装置 设备 | ||
1.一种数据处理方法,其特征在于,所述方法包括:
将已获取的每个无标签图像输入给机器学习模型,由所述机器学习模型输出每个无标签图像对应的目标框和目标类别;
针对每个目标类别,对所述目标类别对应的所有目标框的无标签数据特征进行聚类,得到所述目标类别对应的至少一个无标签聚类中心;
针对每个无标签聚类中心,确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离;基于所述距离确定所述无标签聚类中心的伪标签类型,所述伪标签类型是正报类型或者误报类型;
基于所有目标类别对应的所有无标签聚类中心的伪标签类型,统计正报类型的伪标签类型的第一数量和误报类型的伪标签类型的第二数量;
基于所述第一数量和所述第二数量确定所述机器学习模型是否需要重新训练;若是,则对所述机器学习模型进行重新训练,得到训练后的机器学习模型,并基于所述训练后的机器学习模型对待检测图像进行数据处理。
2.根据权利要求1所述的方法,其特征在于,所述确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离之前,还包括:
针对每个目标类别,对所述目标类别对应的所有标定框的有标签数据特征进行聚类,得到所述目标类别对应的至少一个有标签聚类中心;
其中,已获取的训练数据集中包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别。
3.根据权利要求2所述的方法,其特征在于,
所述对所述目标类别对应的所有标定框的有标签数据特征进行聚类之前,所述方法还包括:针对每个标定框,将所述标定框输入给通用特征提取模型,由所述通用特征提取模型输出所述标定框对应的有标签数据特征;
所述对所述目标类别对应的所有目标框的无标签数据特征进行聚类之前,所述方法还包括:针对每个目标框,将所述目标框输入给通用特征提取模型,由所述通用特征提取模型输出所述目标框对应的无标签数据特征。
4.根据权利要求1所述的方法,其特征在于,若所述目标类别对应M个有标签聚类中心,M为正整数,则距离的总数量为M个;
所述基于所述距离确定所述无标签聚类中心的伪标签类型,包括:
从M个距离中选取出最小距离;若所述最小距离小于第一距离阈值,则确定所述无标签聚类中心的伪标签类型是正报类型;若所述最小距离大于第二距离阈值,则确定所述无标签聚类中心的伪标签类型是误报类型;
其中,所述第二距离阈值大于或等于所述第一距离阈值。
5.根据权利要求1所述的方法,其特征在于,所述基于所述第一数量和所述第二数量确定所述机器学习模型是否需要重新训练,包括:
基于所述第一数量和所述第二数量确定伪标签类型的总数量;
若所述第二数量与所述总数量的比例大于第一比例阈值,则确定所述机器学习模型需要重新训练;若所述第二数量与所述总数量的比例不大于所述第一比例阈值,则确定所述机器学习模型不需要重新训练;
或,若所述第一数量与所述总数量的比例小于第二比例阈值,则确定所述机器学习模型需要重新训练;若所述第一数量与所述总数量的比例不小于所述第二比例阈值,则确定所述机器学习模型不需要重新训练。
6.根据权利要求1所述的方法,其特征在于,所述对所述机器学习模型进行重新训练,得到训练后的机器学习模型,包括:
获取样本数据集,所述样本数据集包括多个无标签图像,并获取训练数据集,所述训练数据集包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别;基于所述样本数据集和所述训练数据集对所述机器学习模型进行重新训练,得到训练后的机器学习模型。
7.根据权利要求6所述的方法,其特征在于,
所述样本数据集包括K1个第一类型的无标签图像和K2个第二类型的无标签图像,所述K1与所述K2相同或者不同;其中,所述第一类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是误报类型;所述第二类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是正报类型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于杭州海康威视数字技术股份有限公司,未经杭州海康威视数字技术股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110956614.7/1.html,转载请声明来源钻瓜专利网。





