一种用于表面缺陷检测的正样本扩充方法

摘要

本发明公开了一种用于表面缺陷检测的正样本扩充方法，包括，将检测器生成的目标框和检测框的位置坐标输入至非极大抑制单元，通过非极大抑制单元去除重叠检测框，保留包含目标信息的检测框；通过样本判定单元筛选和判定负样本、正样本以及半正样本；通过样本采样单元根据设定的正负样本采样比例对负样本、正样本以及半正样本进行采样；将全部采样数据输入至损失回归单元，通过损失回归单元进行检测框分类损失训练和检测框定位回归训练，进而扩充正样本；本发明可以更准确地衡量检测框与目标框之间的重叠程度；同时，改善了正负样本的不平衡问题，且可直接迁移应用至多种目标检测网络中，无需预训练，通用性好。