【技术实现步骤摘要】
一种产品缺陷检测方法和装置
本申请涉及计算机视觉
,尤其涉及一种产品缺陷检测方法和装置。
技术介绍
在产品制造过程中,由于工艺不稳定、厂房环境变化、使用原材料不良等因素的影响,生产出的产品会存在各种形态各异的缺陷、利用计算机视觉技术可以实现对人工检测缺陷方式的替代,但是大多数计算机视觉技术所依赖的产品缺陷检测模型要依赖于大量有标注的训练样本,而这依然需要较多的人工参与,成本高,效率低。
技术实现思路
本申请实施例提供了一种产品缺陷检测方法和装置,可以实现高效的产品缺陷检测,并且依赖的模型可依赖少量训练样本即可训练完成。本申请实施例采用下述技术方案:第一方面,本申请实施例提供一种产品缺陷检测方法,包括:利用Unet网络生成产品图像的掩模图像;根据掩模图像中指定像素值的分布情况确定产品特征;设置与产品特征相对应的缺陷条件;将产品特征与缺陷条件进行匹配,若匹配成功,则产品中存在缺陷。在一些实施例中,根据掩模图像中指定像素值的分布情况确定产品特征包括:确定能够涵盖掩模图像中连通的指定 ...
【技术保护点】
1.一种产品缺陷检测方法,包括:/n利用Unet网络生成产品图像的掩模图像;/n根据所述掩模图像中指定像素值的分布情况确定产品特征;/n设置与所述产品特征相对应的缺陷条件;/n将所述产品特征与缺陷条件进行匹配,若匹配成功,则所述产品中存在缺陷。/n
【技术特征摘要】
1.一种产品缺陷检测方法,包括:
利用Unet网络生成产品图像的掩模图像;
根据所述掩模图像中指定像素值的分布情况确定产品特征;
设置与所述产品特征相对应的缺陷条件;
将所述产品特征与缺陷条件进行匹配,若匹配成功,则所述产品中存在缺陷。
2.如权利要求1所述的方法,其特征在于,所述根据所述掩模图像中指定像素值的分布情况确定产品特征包括:
确定能够涵盖所述掩模图像中连通的指定像素值的定位框,将所述定位框的几何特征作为产品特征;
所述缺陷条件为:所述几何特征的特征值大于预设阈值。
3.如权利要求2所述的方法,其特征在于,所述缺陷特征条件中的预设阈值是根据各训练样本图像中的缺陷大小确定的,从而使预设比例数量的训练样本图像的产品特征能够与所述缺陷特征条件相匹配。
4.如权利要求1所述的方法,其特征在于,所述Unet网络是通过如下方式训练的:
根据训练样本图像的缺陷标注信息,为训练样本图像生成第一掩模图像作为样本真实值,所述第一掩模图像的所述指定像素值与所述缺陷标注信息相对应;
将所述训练样本图像输入到所述Unet网络中,得到所述Unet网络输出的第二掩模图像作为样本预测值;
根据所述样本真实值和所述样本预测值计算训练损失值,根据所述训练损失值进行所述Unet网络的优化。
5.如权利要求4所述的方法,其特征在于,所述根据所述样本真实值和所述样本预测值计算训练...
【专利技术属性】
技术研发人员:张文超,冯扬扬,刘杰,张一凡,
申请(专利权)人:歌尔股份有限公司,
类型:发明
国别省市:山东;37
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。