CN110969200A - 基于一致性负样本的图像目标检测模型训练方法及装置 - Google Patents
基于一致性负样本的图像目标检测模型训练方法及装置 Download PDFInfo
- Publication number
- CN110969200A CN110969200A CN201911183070.4A CN201911183070A CN110969200A CN 110969200 A CN110969200 A CN 110969200A CN 201911183070 A CN201911183070 A CN 201911183070A CN 110969200 A CN110969200 A CN 110969200A
- Authority
- CN
- China
- Prior art keywords
- initial
- image
- frame
- anchor frame
- sample set
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Granted
Links
- 238000001514 detection method Methods 0.000 title claims abstract description 71
- 238000000034 method Methods 0.000 title claims abstract description 47
- 238000012549 training Methods 0.000 title claims abstract description 38
- 238000013528 artificial neural network Methods 0.000 claims description 6
- 238000012545 processing Methods 0.000 abstract description 3
- 230000008569 process Effects 0.000 description 11
- 230000006870 function Effects 0.000 description 7
- 230000006399 behavior Effects 0.000 description 5
- 230000008878 coupling Effects 0.000 description 3
- 238000010168 coupling process Methods 0.000 description 3
- 238000005859 coupling reaction Methods 0.000 description 3
- 238000004891 communication Methods 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 238000005065 mining Methods 0.000 description 2
- 238000012544 monitoring process Methods 0.000 description 2
- 238000013459 approach Methods 0.000 description 1
- 230000008859 change Effects 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000010586 diagram Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000007935 neutral effect Effects 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T10/00—Road transport of goods or passengers
- Y02T10/10—Internal combustion engine [ICE] based vehicles
- Y02T10/40—Engine management systems
Abstract
本发明涉及图像处理技术领域,具体涉及一种基于一致性负样本的图像目标检测模型训练方法及装置。为了解决现有技术采用固定负样本导致模型难以学习到有判别力的特征的问题,本发明提出一种图像目标检测模型训练方法,该方法包括基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集;根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集;通过所述更新图像样本集训练所述图像目标检测模型。利用本发明的方法和装置能够利用更为全面的信息训练图像目标检测模型。
Description
技术领域
本发明涉及图像处理技术领域,具体涉及一种基于一致性负样本的图像目标检测模型训练方法及装置。
背景技术
目标检测是预测图像中所有目标物体的位置,并用矩形框将目标物体的位置进行标记,预测矩形框内物体的类别。目前鲁棒高效的方法大多基于卷积神经网络,常采用基于锚框的方式,即事先在图像上按一定间隔均匀铺满锚框,在预测过程中,模型对先验锚框进行回归,使它们的形状和位置逐渐贴近前景物体,同时模型对框内的物体进行类别预测,从而完成目标检测。
为了对模型进行监督训练,初始平铺在图像中的先验锚框要预先设定各自的回归目标,即对应具体的真实框。回归目标的匹配由锚框和真实框的重叠度决定,并由此确定正负样本。
在以往的方法中,正负样本的分配是一个一次性操作,并由正负分配阈值决定(“正负”针对样本性质而言,非数值正负)。当锚框和其最近真实框的重叠度高于正阈值时为正样本,和所有真实框的重叠度均低于负阈值时为负样本,其它未分配的样本为中性样本且不参与训练。
采用不同的阈值对,会对正负样本集的生成产生影响。虽然不同检测器在阈值的设定上有差别,但它们都使用了固定负样本,即负样本一旦在训练前被分配好,在训练过程中就都是固定不变的。初始正负样本的分配完全基于人类先验,即默认离物体远的框不会回归到物体附近。但恰有某些负样本在经过一次次迭代优化过的模型后产生了和初始不同的表现,即不再处于背景区。由于在深度学习中模型受数据驱动进行优化,如此的有偏负样本会混淆前景和背景的概念,使模型难以学到有判别力的特征。
因此,如何提出一种解决现有技术问题的方案是本领域技术人员需要解决的技术问题。
发明内容
为了解决现有技术中的上述问题,即为了解决现有技术采用固定负样本导致模型难以学习到有判别力的特征的问题,本发明的第一方面提供了一种基于一致性负样本的图像目标检测模型训练方法,所述方法包括:
基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集;
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合;
根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集;
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置;
通过所述更新图像样本集训练所述图像目标检测模型。
优选地,所述真实框的个数为多个,“基于待识别图像中真实框和预设的初始锚框的重叠度,获取初始图像样本集”,其方法包括:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
优选地,所述初始图像样本集包括负样本集合,所述负样本集合包括所述预测锚框,“基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集”,其方法包括:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
优选地,“根据所述一致性负样本获取所述更新图像样本集”,其方法包括:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
优选地,“通过所述更新图像样本集训练所述图像目标检测模型”,其方法包括:
按照如下公式中的目标函数训练所述图像目标检测模型:
其中,所述Lk({cb},{rb})表示所述图像目标检测模型的目标函数,所述表示负样本集合的分类损失和,所述P表示正样本集合,所述b表示初始锚框,所述Lcls表示分类损失,所述cb表示所述初始锚框的预测类别分数,所述表示所述初始锚框的类别真值,所述α表示平衡分类和回归损失的参数,所述Lreg表示回归损失,所述rb表示初始锚框的预测位置,所述表示预测框的位置真值,所述Nk表示经过k次迭代后的负样本集合,所述表示经过k次迭代后的分类损失。
本发明的第二方面还提供了一种图像目标检测模型训练装置,所述装置包括;
第一模块,所述第一模块用于基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集;
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合;
第二模块,所述第二模块用于根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集;
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置;
第三模块,所述第三模块用于通过所述更新图像样本集训练所述图像目标检测模型。
优选地,所述真实框的个数为多个,所述第一模块还用于:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在所述第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
优选地,所述初始图像样本集包括负样本集合,所述负样本集合包括所述预测锚框,所述第二模块还用于:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
优选地,所述第二模块还用于:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
优选地,所述第三模块还用于:
按照如下公式中的目标函数训练所述图像目标检测模型:
其中,所述Lk({cb},{rb})表示所述图像目标检测模型的目标函数,所述表示负样本的分类损失和,所述P表示正样本集合,所述b表示初始锚框,所述Lcls表示分类损失,所述cb表示所述初始锚框的预测类别分数,所述表示所述初始锚框的类别真值,所述α表示平衡分类和回归损失的参数,所述Lreg表示回归损失,所述rb表示初始锚框的预测位置,所述表示预测框的位置真值,所述Nk表示经过k次迭代后的负样本集合,所述表示经过k次迭代后的分类损失。
本发明提供的图像目标检测模型训练方法和装置,能够在挖掘一致性负样本的过程中实现对负样本回归行为的监控以及动态调整;在训练过程中使用一致性负样本,增强了前景和背景的区分能力,能够让图像目标检测模型学习到更具有判别力的特征;此外,由于负样本没有对应的真实位置,而一致性负样本在定义过程中包含了位置信息,再结合负样本的回归行为,从而使得能够利用更为全面的信息训练图像目标检测模型。
附图说明
图1是本发明的基于一致性负样本的图像目标检测模型训练方法的流程示意图;
图2是本发明的基于一致性负样本的图像目标检测模型训练装置的结构示意图。
具体实施方式
为使本发明的实施例、技术方案和优点更加明显,下面将结合附图对本发明的技术方案进行清楚、完整地描述,显然,所述的实施例是本发明的一部分实施例,而不是全部实施例。本领域技术人员应当理解的是,这些实施方式仅仅用于解释本发明的技术原理,并非旨在限制本发明的保护范围。
如图1所示,图1示例性地示出了本申请基于一致性负样本的图像目标检测模型训练方法的流程示意图。本申请基于一致性负样本的图像目标检测模型训练方法包括如下步骤:
步骤S101、基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集。
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合。
可以理解的是,目标检测算法通常会在输入图像中采样大量区域,然后判断所采样区域中是否包含所感兴趣的目标,并相应地调整区域边缘,从而能够更准确地预测目标的真实边界框(ground-truth bounding box)。其中,将以每个像素为中心生成多个大小和宽高比不同的边界框称为锚框(anchor box)。目标检测算法中可以基于锚框实现目标检测。
具体地,基于锚框的目标检测算法中,可以事先在待识别图像中按一定间隔均匀铺满锚框,在对待识别物体进行检测过程中,可以对预先设定的初始锚框进行回归,使得初始锚框的形状和位置更加贴近待识别图像中的前景物体。其中,为了使得图像目标检测模型能够更准确地进行物体识别,需要利用图像样本集对图像目标检测模型进行训练。图像目标检测模型识别的准确度在一定程度上取决于图像样本集中样本选取的合理性。
作为一种实现方式,可以采用如下方法获取初始图像样本集:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在所述第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
需要说明的是,正样本集合中的第一锚框与真实框的重叠度大,所选定的第一锚框的位置往往更加接近真实物体的位置;负样本集合中第二锚框与真实框的重叠度小,所选定的第二锚框的位置往往远离真实物体的位置。此外,负样本集合中的第二锚框的位置在对图像目标检测模型进行训练时,与真实框的重叠度是会产生变化的,可能存在某些第二锚框的位置回归到真实框位置附近,导致图像识别的精度降低。
因此,为了解决有偏负样本对图像目标检测模型训练带来的干扰,需要改善训练样本,使得训练过程有效且能够提高模型的识别精度。
步骤S102、根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集。
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置。
为了保证所训练的图像目标检测模型在实际应用中有鲁棒的前景背景区分能力和检测性能,需要动态挖掘一致性负样本并且更新负样本集。
作为一种实现方式,可以采用如下方法获取更新图像样本集:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
作为一种实现方式,可以采用如下方法获取更新图像样本集:
将一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以更新初始图像样本集;
或者
根据一致性负样本,将初始负样本集合中非一致性负样本的权重设定为0,用以更新初始图像样本集。
可以理解的是,若初始锚框与某个真实框重叠度(intersection-over-union,交并比)大于等于第一预设阈值tpos,则将重叠度大于等于第一预设阈值tpos的初始锚框认定为第一锚框,将第一锚框的集合作为初始正样本集合;相应地,若所有真实框和初始锚框的重叠度均小于第二预设阈值tneg,则将重叠度小于第二预设阈值tneg的初始锚框认定为第二锚框,将第二锚框的集合作为初始负样本集合。
针对每个第二锚框,重新计算其通过图像目标检测模型后输出的结果和真实框的重叠度,若重叠度仍然小于第二预设阈值tneg,则将重叠度仍小于第二预设阈值的第二锚框认定为一致性负样本。
得到一致性负样本后,可以更新初始图像样本集。具体地,可以采用如下方法获取更新图像样本集:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
本发明的目标检测效果优于使用传统正负样本分配方法进行训练的检测器的效果,由于动态调整了有偏负样本集,训练出的检测器对待检测的正负物体区分度更大,提高了检测的召回率和精度;本方法在传统样本分配方法的基础上进行再分配,没有引入额外参数,且只作用在训练过程,所以不需要更改原检测器的结构,使本方法在不同场景下具有泛化性和可迁移性。
步骤S103、通过所述更新图像样本集训练所述图像目标检测模型。
具体地,可以按照如下公式(1)目标函数训练图像目标检测模型:
公式(1):
其中,Lk({cb},{rb})表示图像目标检测模型的目标函数,表示负样本的分类损失和,P表示正样本集合,b表示初始锚框,Lcls表示分类损失,cb表示初始锚框的预测类别分数,表示初始锚框的类别真值,α表示平衡分类和回归损失的参数,Lreg表示回归损失,rb表示初始锚框的预测位置,表示预测框的位置真值,Nk表示经过k次迭代后的负样本集合,表示经过k次迭代后的分类损失。
本发明提供的基于一致性负样本的图像目标检测模型训练方法能够在挖掘一致性负样本的过程中实现对负样本回归行为的监控以及动态调整;在训练过程中使用一致性负样本,增强了前景和背景的区分能力,能够让图像目标检测模型学习到更具有判别力的特征;此外,由于负样本没有对应的真实位置,而一致性负样本在定义过程中包含了位置信息,再结合负样本的回归行为,从而使得能够利用更为全面的信息训练图像目标检测模型。
参照上述方法实施例,本发明还提供了一种基于一致性负样本的图像目标检测模型训练装置,装置包括:
第一模块1,第一模块1用于基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集,
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合;
第二模块2,第二模块2用于根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集,
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置;
第三模块3,第三模块3用于通过所述更新图像样本集训练所述图像目标检测模型。
优选地,第一模块1还用于:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在所述第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
优选地,所述初始图像样本集包括负样本集合,所述负样本集合包括所述预测锚框,第二模块2还用于:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
优选地,第二模块2还用于:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
优选地,第三模块3还用于:
按照公式(1)中的目标函数训练所述图像目标检测模型。
在本申请所提供的几个实施例中,应该理解到,所揭露的系统,装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述模块或单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本申请各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的全部或部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)或处理器(processor)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
综上所述,以上实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围。
Claims (10)
1.一种基于一致性负样本的图像目标检测模型训练方法,其特征在于,所述方法包括:
基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集;
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合;
根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集;
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置;
通过所述更新图像样本集训练所述图像目标检测模型。
2.根据权利要求1所述的方法,其特征在于,所述真实框的个数为多个,“基于待识别图像中真实框和预设的初始锚框的重叠度,获取初始图像样本集”,其方法包括:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在所述第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
3.根据权利要求2所述的方法,其特征在于,所述初始图像样本集包括负样本集合,所述负样本集合包括所述预测锚框,“基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集”,其方法包括:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
4.根据权利要求3所述的方法,其特征在于,“根据所述一致性负样本获取所述更新图像样本集”,其方法包括:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
6.一种基于一致性负样本的图像目标检测模型训练装置,其特征在于,所述装置包括:
第一模块,所述第一模块用于基于待识别图像中的真实框和预设的初始锚框的重叠度,获取初始图像样本集;
其中,所述初始锚框为所述待识别图像中预先设定的锚框,所述真实框为所述待识别图像中待识别物体对应的真实矩形框,所述初始图像样本集为与所述真实框的重叠度满足预设条件的初始锚框的集合;
第二模块,所述第二模块用于根据所述初始图像样本集,通过预设的图像目标检测模型,获取所述与初始锚框对应的预测锚框,基于所述真实框与所述预测锚框的重叠度,获取更新图像样本集;
其中,所述图像目标检测模型基于神经网络构建,通过预设图像样本集训练,用于识别图像中待识别物体和确定所述待识别物体的真实框的位置;
第三模块,所述第三模块用于通过所述更新图像样本集训练所述图像目标检测模型。
7.根据权利要求6所述的装置,其特征在于,所述真实框的个数为多个,所述第一模块还用于:
分别获取每个所述初始锚框与多个所述真实框的第一重叠度,
若存在所述第一重叠度大于等于第一预设阈值,则将与至少一个所述真实框的第一重叠度大于等于第一预设阈值的初始锚框认定为第一锚框,将所述第一锚框的集合作为正样本集合;
若每个所述第一重叠度均小于第二预设阈值,则将与多个所述真实框的第一重叠度均小于第二预设阈值的初始锚框认定为第二锚框,将所述第二锚框的集合作为初始负样本集合;
根据所述正样本集合和所述初始负样本集合确定所述初始图像样本集。
8.根据权利要求7所述的装置,其特征在于,所述初始图像样本集包括负样本集合,所述负样本集合包括所述预测锚框,所述第二模块还用于:
获取每个所述预测锚框与多个所述真实框的第二重叠度;
若每个所述第二重叠度仍小于所述第二预设阈值,则将每个所述第二重叠度仍小于所述第二预设阈值的预测锚框认定为一致性负样本;
根据所述一致性负样本获取所述更新图像样本集。
9.根据权利要求8所述的装置,其特征在于,所述第二模块还用于:
将所述一致性负样本的集合替换初始图像样本集中的初始负样本集合,用以获取所述更新图像样本集;
或者
根据所述一致性负样本,将所述初始负样本集合中非一致性负样本的权重设定为0,用以获取所述更新图像样本集。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201911183070.4A CN110969200B (zh) | 2019-11-27 | 2019-11-27 | 基于一致性负样本的图像目标检测模型训练方法及装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201911183070.4A CN110969200B (zh) | 2019-11-27 | 2019-11-27 | 基于一致性负样本的图像目标检测模型训练方法及装置 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN110969200A true CN110969200A (zh) | 2020-04-07 |
CN110969200B CN110969200B (zh) | 2023-05-12 |
Family
ID=70031837
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201911183070.4A Active CN110969200B (zh) | 2019-11-27 | 2019-11-27 | 基于一致性负样本的图像目标检测模型训练方法及装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN110969200B (zh) |
Cited By (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112633496A (zh) * | 2020-12-18 | 2021-04-09 | 杭州海康威视数字技术股份有限公司 | 一种检测模型的处理方法及装置 |
CN113052217A (zh) * | 2021-03-15 | 2021-06-29 | 上海云从汇临人工智能科技有限公司 | 预测结果标识及其模型训练方法、装置及计算机存储介质 |
CN113239982A (zh) * | 2021-04-23 | 2021-08-10 | 北京旷视科技有限公司 | 检测模型的训练方法、目标检测方法、装置和电子系统 |
CN114862683A (zh) * | 2022-07-07 | 2022-08-05 | 浪潮电子信息产业股份有限公司 | 模型生成方法、目标检测方法、装置、设备及介质 |
CN115409818A (zh) * | 2022-09-05 | 2022-11-29 | 江苏济远医疗科技有限公司 | 一种应用于内窥镜图像目标检测模型的增强训练方法 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20170206431A1 (en) * | 2016-01-20 | 2017-07-20 | Microsoft Technology Licensing, Llc | Object detection and classification in images |
CN109543740A (zh) * | 2018-11-14 | 2019-03-29 | 哈尔滨工程大学 | 一种基于生成对抗网络的目标检测方法 |
CN110443212A (zh) * | 2019-08-12 | 2019-11-12 | 睿魔智能科技(深圳)有限公司 | 用于目标检测的正样本获取方法、装置、设备及存储介质 |
-
2019
- 2019-11-27 CN CN201911183070.4A patent/CN110969200B/zh active Active
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20170206431A1 (en) * | 2016-01-20 | 2017-07-20 | Microsoft Technology Licensing, Llc | Object detection and classification in images |
CN109543740A (zh) * | 2018-11-14 | 2019-03-29 | 哈尔滨工程大学 | 一种基于生成对抗网络的目标检测方法 |
CN110443212A (zh) * | 2019-08-12 | 2019-11-12 | 睿魔智能科技(深圳)有限公司 | 用于目标检测的正样本获取方法、装置、设备及存储介质 |
Non-Patent Citations (1)
Title |
---|
彭刚;杨诗琪;黄心汉;苏豪;: "改进的基于区域卷积神经网络的微操作系统目标检测方法", 模式识别与人工智能 * |
Cited By (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112633496A (zh) * | 2020-12-18 | 2021-04-09 | 杭州海康威视数字技术股份有限公司 | 一种检测模型的处理方法及装置 |
CN112633496B (zh) * | 2020-12-18 | 2023-08-08 | 杭州海康威视数字技术股份有限公司 | 一种检测模型的处理方法及装置 |
CN113052217A (zh) * | 2021-03-15 | 2021-06-29 | 上海云从汇临人工智能科技有限公司 | 预测结果标识及其模型训练方法、装置及计算机存储介质 |
CN113239982A (zh) * | 2021-04-23 | 2021-08-10 | 北京旷视科技有限公司 | 检测模型的训练方法、目标检测方法、装置和电子系统 |
CN114862683A (zh) * | 2022-07-07 | 2022-08-05 | 浪潮电子信息产业股份有限公司 | 模型生成方法、目标检测方法、装置、设备及介质 |
CN115409818A (zh) * | 2022-09-05 | 2022-11-29 | 江苏济远医疗科技有限公司 | 一种应用于内窥镜图像目标检测模型的增强训练方法 |
CN115409818B (zh) * | 2022-09-05 | 2023-10-27 | 江苏济远医疗科技有限公司 | 一种应用于内窥镜图像目标检测模型的增强训练方法 |
Also Published As
Publication number | Publication date |
---|---|
CN110969200B (zh) | 2023-05-12 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN110969200A (zh) | 基于一致性负样本的图像目标检测模型训练方法及装置 | |
CN109117831B (zh) | 物体检测网络的训练方法和装置 | |
CN109272016B (zh) | 目标检测方法、装置、终端设备和计算机可读存储介质 | |
CN111723815B (zh) | 模型训练方法、图像处理方法、装置、计算机系统和介质 | |
CN108830385B (zh) | 深度学习模型训练方法和装置及计算机可读存储介质 | |
CN110889421A (zh) | 目标物检测方法及装置 | |
CN112669275A (zh) | 基于YOLOv3算法的PCB表面缺陷检测方法及装置 | |
CN108805174A (zh) | 聚类方法及装置 | |
CN109685805B (zh) | 一种图像分割方法及装置 | |
CN114330499A (zh) | 分类模型的训练方法、装置、设备、存储介质及程序产品 | |
CN113610069A (zh) | 基于知识蒸馏的目标检测模型训练方法 | |
CN112420125A (zh) | 分子属性预测方法、装置、智能设备和终端 | |
CN115937703A (zh) | 一种用于遥感图像目标检测的增强特征提取方法 | |
CN113822144A (zh) | 一种目标检测方法、装置、计算机设备和存储介质 | |
CN113780287A (zh) | 一种多深度学习模型的最优选取方法及系统 | |
CN111161789B (zh) | 一种模型预测的关键区域的分析方法及装置 | |
CN117371511A (zh) | 图像分类模型的训练方法、装置、设备及存储介质 | |
CN114943674A (zh) | 瑕疵检测方法、电子装置及存储介质 | |
CN115984640B (zh) | 一种基于组合蒸馏技术的目标检测方法、系统和存储介质 | |
CN116451081A (zh) | 数据漂移的检测方法、装置、终端及存储介质 | |
CN116645719A (zh) | 一种瞳孔和虹膜定位方法、装置、电子设备及存储介质 | |
CN114241411B (zh) | 基于目标检测的计数模型处理方法、装置及计算机设备 | |
CN114202224B (zh) | 用于检测生产环境中焊接质量的方法、设备、介质 | |
CN115296984A (zh) | 异常网络节点的检测方法及装置、设备、存储介质 | |
CN117083621A (zh) | 检测器训练方法、装置及存储介质 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |