物体异常位置检测方法、装置、电子设备及存储介质与流程

文档序号:32434897发布日期:2022-12-06 17:45阅读:来源:国知局

技术特征:
1.一种物体异常位置检测方法,其特征在于,所述方法包括:获取目标物体的原始图像;调用预设的目标生成器根据所述原始图像生成所述目标物体的复原图像;根据所述复原图像和所述原始图像,确定所述原始图像的异常信息,所述异常信息用于指示所述目标物体的异常位置。2.根据权利要求1所述的物体异常位置检测方法,其特征在于,所述目标生成器基于预设的生成式对抗网络训练得到,所述生成式对抗网络包括生成器和判别器,所述目标生成器通过如下步骤训练得到:获取训练数据集,所述训练数据集包括至少第一训练样本和第二训练样本;使用所述第一训练样本对所述生成器的网络参数进行调整,直至符合预设的第一停止训练条件时,将所述生成器作为预训练生成器,所述第一训练样本包括多个样本物体的有缺陷真实图像;使用所述第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整,直至符合预设的第二停止训练条件时,将所述预训练生成器作为所述目标生成器,其中,所述第二训练样本包括多个样本物体的有缺陷真实图像。3.根据权利要求2所述的物体异常位置检测方法,其特征在于,在使用第一训练样本对所述生成器的网络参数进行调整过程中,所述生成式对抗网络的训练损失为所述生成器的第一生成损失;在使用第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整过程中,所述生成式对抗网络的训练损失包括所述生成器的第二生成损失和所述判别器的对抗损失;所述使用第一训练样本对所述生成器的网络参数进行调整,直至符合预设的第一停止训练条件时,将所述生成器作为预训练生成器,包括:根据所述生成器基于所述第一训练样本中的有缺陷真实图像生成的预测修复图像,确定所述第一生成损失;根据所述第一生成损失对所述生成器的网络参数进行调整,直至符合预设的第一停止训练条件时,将所述生成器作为预训练生成器;所述使用第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整,直至符合预设的第二停止训练条件时,将所述预训练生成器作为所述目标生成器,包括:根据所述预训练生成器基于所述第二训练样本中的有缺陷真实图像生成的预测修复图像,确定所述第二生成损失和所述对抗损失;根据所述第二生成损失和所述对抗损失,对所述预训练生成器的网络参数和所述判别器的网络参数进行调整,直至符合预设的第二停止训练条件时,将所述预训练生成器作为所述目标生成器。4.根据权利要求3所述的物体异常位置检测方法,其特征在于,所述第一训练样本包括至少一对样本图像,每对所述样本图像包括样本物体的有缺陷真实图像和无缺陷真实图像,所述无缺陷真实图像作为所述有缺陷真实图像的标签,所述根据所述生成器基于所述第一训练样本中的有缺陷真实图像生成的预测修复图像,确定所述第一生成损失,包括:调用所述生成器根据所述有缺陷真实图像生成其对应的预测修复图像;
根据所述预测修复图像、所述无缺陷真实图像和预设的第一损失函数,确定所述第一生成损失。5.根据权利要求3所述的物体异常位置检测方法,其特征在于,所述第二训练样本包括至少一对样本图像,每对所述样本图像包括样本物体的有缺陷真实图像和无缺陷真实图像,所述无缺陷真实图像作为所述有缺陷真实图像的标签,所述根据所述预训练生成器基于所述第二训练样本中的有缺陷真实图像生成的预测修复图像,确定所述第二生成损失和所述对抗损失,包括:调用所述预训练生成器根据所述有缺陷真实图像生成其对应的预测修复图像;调用所述判别器确定所述预测修复图像的判别结果,所述判别结果用于指示所述预测修复图像是否为真实图像;根据所述预测修复图像、所述无缺陷真实图像和预设的第二损失函数,确定所述第二生成损失;根据所述判别结果和预设的第三损失函数,确定所述对抗损失。6.根据权利要求2所述的物体异常位置检测方法,其特征在于,所述生成式对抗网络的生成器为unet网络,所述生成式对抗网络的判别器为卷积神经网络结构。7.根据权利要求1所述的物体异常位置检测方法,其特征在于,所述目标物体是目标pcb板,所述原始图像是所述目标pcb板的缺陷图像,所述调用预设的目标生成器根据所述原始图像生成所述目标物体的复原图像,包括:调用所述目标生成器根据所述缺陷图像生成所述目标pcb板的复原图像;所述根据所述复原图像和所述原始图像,确定所述原始图像的异常信息,包括:根据所述复原图像和所述缺陷图像,确定所述缺陷图像的异常信息,所述异常信息用于指示所述目标pcb的异常位置。8.根据权利要求1-7任一项所述的物体异常位置检测方法,其特征在于,所述根据所述复原图像和所述原始图像,确定所述原始图像的异常信息,包括:根据所述复原图像和所述原始图像,确定所述目标物体的异常位置;根据所述异常位置,确定所述原始图像的异常信息,所述异常信息包括以下至少一种信息:所述异常位置的坐标信息、所述异常位置的区域指示框和所述异常位置的区域图像。9.根据权利要求8所述的物体异常位置检测方法,其特征在于,所述根据所述复原图像和所述原始图像,确定所述目标物体的异常位置,包括:对比所述复原图像和所述原始图像,得到所述原始图像的差异像素点;将所述差异像素点的所在位置作为所述目标物体的异常位置。10.一种物体异常位置检测装置,其特征在于,所述物体异常位置检测装置包括:获取单元,用于获取目标物体的原始图像;生成单元,用于调用预设的目标生成器根据所述原始图像生成所述目标物体的复原图像;确定单元,用于根据所述复原图像和所述原始图像,确定所述原始图像的异常信息,所述异常信息用于指示所述目标物体的异常位置。11.一种电子设备,其特征在于,包括处理器和存储器,所述存储器中存储有计算机程序,所述处理器调用所述存储器中的计算机程序时执行如权利要求1至9任一项所述的物体
异常位置检测方法。12.一种计算机可读存储介质,其特征在于,其上存储有计算机程序,所述计算机程序被处理器进行加载,以执行权利要求1至9任一项所述的物体异常位置检测方法中的步骤。

技术总结
本申请提供一种物体异常位置检测方法、装置、电子设备及计算机可读存储介质。该物体异常位置检测方法包括:获取目标物体的原始图像;调用预设的目标生成器根据所述原始图像生成所述目标物体的复原图像;根据所述复原图像和所述原始图像,确定所述原始图像的异常信息,所述异常信息用于指示所述目标物体的异常位置。本申请中避免了用于异常位置检测的目标检测模型泛化能力差的问题,减少了对样本异常位置标注的时间,降低了训练时间成本。降低了训练时间成本。降低了训练时间成本。


技术研发人员:李秀阳
受保护的技术使用者:武汉TCL集团工业研究院有限公司
技术研发日:2021.06.02
技术公布日:2022/12/5
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1