利索能及
我要发布
收藏
专利号: 2021109566147
申请人: 杭州海康威视数字技术股份有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2025-08-05
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种数据处理方法,其特征在于,所述方法包括:

将已获取的每个无标签图像输入给机器学习模型,由所述机器学习模型输出每个无标签图像对应的目标框和目标类别;

针对每个目标类别,对所述目标类别对应的所有目标框的无标签数据特征进行聚类,得到所述目标类别对应的至少一个无标签聚类中心;

针对所述目标类别对应的每个无标签聚类中心,确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离;基于所述距离确定所述无标签聚类中心的伪标签类型,所述伪标签类型是正报类型或者误报类型;

基于所有目标类别对应的所有无标签聚类中心的伪标签类型,统计正报类型的伪标签类型的第一数量和误报类型的伪标签类型的第二数量;

基于所述第一数量和所述第二数量确定所述机器学习模型是否需要重新训练;若是,则对所述机器学习模型进行重新训练,得到训练后的机器学习模型,并基于所述训练后的机器学习模型对待检测图像进行数据处理。

2.根据权利要求1所述的方法,其特征在于,所述确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离之前,还包括:针对每个目标类别,对所述目标类别对应的所有标定框的有标签数据特征进行聚类,得到所述目标类别对应的至少一个有标签聚类中心;

其中,已获取的训练数据集中包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别。

3.根据权利要求2所述的方法,其特征在于,

所述对所述目标类别对应的所有标定框的有标签数据特征进行聚类之前,所述方法还包括:针对每个标定框,将所述标定框输入给通用特征提取模型,由所述通用特征提取模型输出所述标定框对应的有标签数据特征;

所述对所述目标类别对应的所有目标框的无标签数据特征进行聚类之前,所述方法还包括:针对每个目标框,将所述目标框输入给通用特征提取模型,由所述通用特征提取模型输出所述目标框对应的无标签数据特征。

4.根据权利要求1所述的方法,其特征在于,若所述目标类别对应M个有标签聚类中心,M为正整数,则距离的总数量为M个;

所述基于所述距离确定所述无标签聚类中心的伪标签类型,包括:

从M个距离中选取出最小距离;若所述最小距离小于第一距离阈值,则确定所述无标签聚类中心的伪标签类型是正报类型;若所述最小距离大于第二距离阈值,则确定所述无标签聚类中心的伪标签类型是误报类型;

其中,所述第二距离阈值大于或等于所述第一距离阈值。

5.根据权利要求1所述的方法,其特征在于,所述基于所述第一数量和所述第二数量确定所述机器学习模型是否需要重新训练,包括:基于所述第一数量和所述第二数量确定伪标签类型的总数量;

若所述第二数量与所述总数量的比例大于第一比例阈值,则确定所述机器学习模型需要重新训练;若所述第二数量与所述总数量的比例不大于所述第一比例阈值,则确定所述机器学习模型不需要重新训练;

或,若所述第一数量与所述总数量的比例小于第二比例阈值,则确定所述机器学习模型需要重新训练;若所述第一数量与所述总数量的比例不小于所述第二比例阈值,则确定所述机器学习模型不需要重新训练。

6.根据权利要求1所述的方法,其特征在于,所述对所述机器学习模型进行重新训练,得到训练后的机器学习模型,包括:获取样本数据集,所述样本数据集包括多个无标签图像,并获取训练数据集,所述训练数据集包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别;基于所述样本数据集和所述训练数据集对所述机器学习模型进行重新训练,得到训练后的机器学习模型。

7.根据权利要求6所述的方法,其特征在于,

所述样本数据集包括K1个第一类型的无标签图像和K2个第二类型的无标签图像,所述K1与所述K2相同或者不同;其中,所述第一类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是误报类型;所述第二类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是正报类型。

8.一种数据处理装置,其特征在于,所述装置包括:

获取模块,用于将已获取的每个无标签图像输入给机器学习模型,由所述机器学习模型输出每个无标签图像对应的目标框和目标类别;

聚类模块,用于针对每个目标类别,对所述目标类别对应的所有目标框的无标签数据特征进行聚类,得到所述目标类别对应的至少一个无标签聚类中心;

确定模块,用于针对所述目标类别对应的每个无标签聚类中心,确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离;基于所述距离确定所述无标签聚类中心的伪标签类型,所述伪标签类型是正报类型或误报类型;

统计模块,用于基于所有目标类别对应的所有无标签聚类中心的伪标签类型统计正报类型的伪标签类型的第一数量和误报类型的伪标签类型的第二数量;

训练模块,用于基于第一数量和第二数量确定机器学习模型是否需要重新训练;若是,对所述机器学习模型进行重新训练,得到训练后的机器学习模型;

处理模块,用于基于训练后的机器学习模型对待检测图像进行数据处理。

9.根据权利要求8所述的装置,其特征在于,

所述聚类模块,还用于针对每个目标类别,对所述目标类别对应的所有标定框的有标签数据特征进行聚类,得到所述目标类别对应的至少一个有标签聚类中心;其中,已获取的训练数据集中包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别;

所述获取模块,还用于在对所述目标类别对应的所有标定框的有标签数据特征进行聚类之前,针对每个标定框,将所述标定框输入给通用特征提取模型,由所述通用特征提取模型输出所述标定框对应的有标签数据特征;

所述获取模块,还用于在对所述目标类别对应的所有目标框的无标签数据特征进行聚类之前,针对每个目标框,将所述目标框输入给通用特征提取模型,由所述通用特征提取模型输出所述目标框对应的无标签数据特征;

若所述目标类别对应M个有标签聚类中心,所述M为正整数,则距离的总数量为M个;所述确定模块基于所述距离确定所述无标签聚类中心的伪标签类型时具体用于:从M个距离中选取出最小距离;若所述最小距离小于第一距离阈值,则确定所述无标签聚类中心的伪标签类型是正报类型;若所述最小距离大于第二距离阈值,则确定所述无标签聚类中心的伪标签类型是误报类型;其中,所述第二距离阈值大于或等于所述第一距离阈值;

所述训练模块基于第一数量和第二数量确定机器学习模型是否需要重新训练时具体用于:基于第一数量和第二数量确定伪标签类型的总数量;若第二数量与所述总数量的比例大于第一比例阈值,则确定所述机器学习模型需要重新训练;若第二数量与所述总数量的比例不大于第一比例阈值,则确定所述机器学习模型不需要重新训练;或,若第一数量与所述总数量的比例小于第二比例阈值,则确定所述机器学习模型需要重新训练;若第一数量与所述总数量的比例不小于第二比例阈值,则确定所述机器学习模型不需要重新训练;

所述训练模块对所述机器学习模型进行重新训练,得到训练后的机器学习模型时具体用于:获取样本数据集,所述样本数据集包括多个无标签图像,获取训练数据集,所述训练数据集包括多个有标签图像,针对每个有标签图像,所述有标签图像对应有标定框和目标类别;基于所述样本数据集和所述训练数据集对所述机器学习模型进行重新训练,得到训练后的机器学习模型;其中,所述样本数据集包括K1个第一类型的无标签图像和K2个第二类型的无标签图像,所述K1与所述K2相同或不同;所述第一类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是误报类型;所述第二类型的无标签图像中的目标框所属的无标签聚类中心的伪标签类型是正报类型。

10.一种数据处理设备,其特征在于,包括:处理器和机器可读存储介质,所述机器可读存储介质存储有能够被所述处理器执行的机器可执行指令;

所述处理器用于执行机器可执行指令,以实现如下步骤:

将已获取的每个无标签图像输入给机器学习模型,由所述机器学习模型输出每个无标签图像对应的目标框和目标类别;

针对每个目标类别,对所述目标类别对应的所有目标框的无标签数据特征进行聚类,得到所述目标类别对应的至少一个无标签聚类中心;

针对所述目标类别对应的每个无标签聚类中心,确定所述无标签聚类中心与所述目标类别对应的已获取的有标签聚类中心之间的距离;基于所述距离确定所述无标签聚类中心的伪标签类型,所述伪标签类型是正报类型或者误报类型;

基于所有目标类别对应的所有无标签聚类中心的伪标签类型,统计正报类型的伪标签类型的第一数量和误报类型的伪标签类型的第二数量;

基于所述第一数量和所述第二数量确定所述机器学习模型是否需要重新训练;若是,则对所述机器学习模型进行重新训练,得到训练后的机器学习模型,并基于所述训练后的机器学习模型对待检测图像进行数据处理。