[发明专利]异常检测方法、装置、电子设备及计算机可读存储介质

说明书

本公开实施例公开了一种异常检测方法、装置、电子设备及计算机可读存储介质。该方法包括：采用第一样本集对初始检测网络进行训练，得到检测网络；所述第一样本集为正样本集；在所述检测网络对多张待检测图像进行异常检测过程中，基于检测出的正常图像集和异常图像集，得到第二样本集，所述第二样本集为包含正样本和负样本的增量样本集；其中，所述负样本为图像中存在异常的异常图像；采用所述第二样本集，对所述检测网络进行更新训练，得到更新后的检测网络。通过本公开，能够提高异常检测的准确性与灵活性。

技术领域

本公开涉及机器视觉技术，尤其涉及一种异常检测方法、装置、电子设备及计算机可读存储介质。

背景技术

近年来，深度学习算法在各个领域都取得了巨大的进展，也在很多工业视觉检测领域取得了落地。相关技术通过深度学习算法训练得到异常检测模型用于异常图像的检测。深度学习算法往往需要大量的、高质量的训练样本，然而，目前基于正常样本学习方式与基于异常样本学习方式所适用的模型结构以及学习策略不同，因此目前的深度学习方法对于不同类型的训练样本的兼容能力较差，导致所能使用的训练样本较为单一，这样不仅降低了异常检测的灵活性，也降低了异常检测的准确性。

发明内容

本公开实施例提供一种异常检测方法、装置、电子设备及计算机可读存储介质，能够提高异常检测的准确性与灵活性。

本公开实施例的技术方案是这样实现的：

本公开实施例提供一种异常检测方法，包括：采用第一样本集对初始检测网络进行训练，得到检测网络；所述第一样本集为正样本集；在所述检测网络对多张待检测图像进行异常检测过程中，基于检测出的正常图像集和异常图像集，得到第二样本集，所述第二样本集为包含正样本和负样本的增量样本集；其中，所述负样本为图像中存在异常的异常图像；采用所述第二样本集，对所述检测网络进行更新训练，得到更新后的检测网络。

上述方法中，所述采用第一正样本集对初始检测网络进行训练，得到检测网络，包括：采用所述初始检测网络对所述第一样本集中的正样本进行检测，得到所述正样本对应的第一特征差异序列；根据所述第一特征差异序列和正常损失函数，确定出所述正样本对应的训练损失；基于所述训练损失，对所述初始检测网络进行训练调整，直至得到的最终损失小于预设损失阈值时，得到所述检测网络；其中，所述正常损失函数表征所述正样本对应的重构特征序列拉近所述正样本。

上述方法中，所述基于检测出的正常图像集和异常图像集，得到第二样本集，包括：从检测出的所述正常图像集和所述异常图像集中，确定待核验图像；对所述待核验图像进行核验，得到所述第二样本集。

上述方法中，所述待核验图像中包括：至少一张异常图像和至少一张正常图像；所述对所述待核验图像进行核验，得到所述第二样本集，包括：对所述至少一张异常图像和所述至少一张正常图像分别进行核验，得到各自的核验结果；将所述核验结果表征正确的异常图像作为所述负样本，并将所述核验结果表征正确的正常图像作为所述正样本；对所述核验结果表征错误的异常图像，以及所述核验结果表征错误的正常图像进行正确标注，得到所述正样本和所述负样本；将所述负样本和所述正样本的集合，确定为所述第二样本集。

上述方法中，所述采用第一样本集对初始检测网络进行训练，得到检测网络之后，所述方法还包括：采用所述检测网络对每张待检测图像进行异常检测，得到所述每张待检测图像的每个像素位置上的异常评分；基于所述异常评分，绘制所述每张待检测图像对应的异常检测图像；根据所述异常检测图像，得到多张待检测图像中的所述正常图像集和所述异常图像集。