[发明专利]物体异常位置检测方法、装置、电子设备及存储介质

说明书

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

技术领域

本申请涉及计算机视觉技术领域，具体涉及一种物体异常位置检测方法、装置、电子设备及计算机可读存储介质。

背景技术

随着计算机视觉技术的飞速发展，计算机视觉技术已经广泛应用于安防监控、医疗健康、工业等众多领域。其中，物体的异常位置检测，比如工业上PCB板的缺陷检测，通常是对样本的异常位置进行标注后，对目标检测模型，比如基于Faster-RCNN系列和YOLO系列的模型进行训练，再采用训练后的目标检测模型来实现目标检测任务。

但是，现有的目标检测模型，例如基于Faster-RCNN系列和YOLO系列的目标检测模型泛化能力差；并且，由于需要依赖大量的标注样本才能训练出效果较好的检测模型，因此训练过程中需要对样本的异常位置进行标注，而大量的样本异常位置标注需要较大的时间成本。

发明内容

本申请提供一种物体异常位置检测方法、装置、电子设备及计算机可读存储介质，旨在解决现有用于异常位置检测的目标检测模型泛化能力差、训练时间成本高的问题。

第一方面，本申请提供一种物体异常位置检测方法，所述方法包括：

获取目标物体的原始图像；

调用预设的目标生成器根据所述原始图像生成所述目标物体的复原图像；

根据所述复原图像和所述原始图像，确定所述原始图像的异常信息，所述异常信息用于指示所述目标物体的异常位置。

在本申请一种可能的实现方式中，所述目标生成器基于预设的生成式对抗网络训练得到，所述生成式对抗网络包括生成器和判别器，所述目标生成器通过如下步骤训练得到：

获取训练数据集，所述训练数据集包括至少第一训练样本和第二训练样本；

使用所述第一训练样本对所述生成器的网络参数进行调整，直至符合预设的第一停止训练条件时，将所述生成器作为预训练生成器，所述第一训练样本包括多个样本物体的有缺陷真实图像；

使用所述第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整，直至符合预设的第二停止训练条件时，将所述预训练生成器作为所述目标生成器，其中，所述第二训练样本包括多个样本物体的有缺陷真实图像。

在本申请一种可能的实现方式中，在使用第一训练样本对所述生成器的网络参数进行调整过程中，所述生成式对抗网络的训练损失为所述生成器的第一生成损失；在使用第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整过程中，所述生成式对抗网络的训练损失包括所述生成器的第二生成损失和所述判别器的对抗损失；

所述使用第一训练样本对所述生成器的网络参数进行调整，直至符合预设的第一停止训练条件时，将所述生成器作为预训练生成器，包括：

根据所述生成器基于所述第一训练样本中的有缺陷真实图像生成的预测修复图像，确定所述第一生成损失；

根据所述第一生成损失对所述生成器的网络参数进行调整，直至符合预设的第一停止训练条件时，将所述生成器作为预训练生成器；

所述使用第二训练样本对所述预训练生成器的网络参数和所述判别器的网络参数进行调整，直至符合预设的第二停止训练条件时，将所述预训练生成器作为所述目标生成器，包括：