[发明专利]图像真伪检测方法、图像真伪检测模型训练方法

说明书

本申请涉及人工智能领域，具体涉及一种图像真伪检测方法。所述方法包括：根据待检测图像中的部分频域信息得到频域缺失图，根据部分色彩信息得到色彩缺失图；根据频域缺失图中的第一图像特征生成频域重建图，根据色彩缺失图中的第二图像特征生成色彩重建图；根据频域重建图与待检测图像之间的差异，确定频域差异掩码，通过频域差异掩码对第一图像特征进行处理，得到频域疑似伪造特征；根据色彩重建图与待检测图像之间的差异，确定色彩差异掩码，通过色彩差异掩码对第二图像特征进行处理，得到色彩疑似伪造特征；综合频域疑似伪造特征和色彩疑似伪造特征，确定待检测图像的真伪检测结果。采用本方法能够提升真伪检测结果的准确性。

技术领域

本申请涉及计算机技术领域，特别是涉及一种图像真伪检测方法、图像真伪检测模型训练方法。

背景技术

目前随着图像编辑技术的发展，人们可以随意编辑图像中的内容，从而导致越来越多以假乱真的高质量伪造图像涌现出来，这将严重影响目前落地的图像识别系统的安全性。因此，检测输入图像是否经过了编辑，日趋成为网络安全中的重要环节。

目前，主要是通过机器学习模型来检测图像中的某些特定伪造痕迹，通过检测出的特定伪造痕迹来判定待检测图像是否为伪造图像。比如，可以通过检测图像中的全局光照和局部光照的匹配程度，来对图像的真伪进行判断。但是通过检测某些特定伪造痕迹来判断待检测图像是否为伪造图像，容易导致机器学习模型的泛化性低，从而使得基于低泛化性的机器学习模型输出的真伪检测结果的准确性低。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提升检测准确定的图像真伪检测方法、装置、计算机设备和存储介质。

一种图像真伪检测方法，所述方法包括：

获取待检测图像，并根据所述待检测图像中的部分频域信息得到频域缺失图，以及根据所述待检测图像中的部分色彩信息得到色彩缺失图；

根据所述频域缺失图中的第一图像特征生成频域重建图，及根据所述色彩缺失图中的第二图像特征生成色彩重建图；

根据所述频域重建图与所述待检测图像之间的差异，确定频域差异掩码，并通过所述频域差异掩码对所述第一图像特征进行自注意力处理，得到频域疑似伪造特征；

根据所述色彩重建图与所述待检测图像之间的差异，确定色彩差异掩码，并通过所述色彩差异掩码对所述第二图像特征进行自注意力处理，得到色彩疑似伪造特征；

综合所述频域疑似伪造特征和色彩疑似伪造特征，确定所述待检测图像的真伪检测结果。

在一个实施例中，所述获取待检测图像，包括：

获取待检测视频，并根据采样频率，提取所述待检测视频中的多个视频帧；

将提取出的每个视频帧均作为待检测图像；

所述方法还包括：

综合每个所述待检测图像分别对应的真伪检测结果，确定所述待检测视频的真伪检测结果。

一种图像真伪检测装置，所述装置包括：

重建图像生成模块，用于获取待检测图像，并根据所述待检测图像中的部分频域信息得到频域缺失图，以及根据所述待检测图像中的部分色彩信息得到色彩缺失图；根据所述频域缺失图中的第一图像特征生成频域重建图，及根据所述色彩缺失图中的第二图像特征生成色彩重建图；

疑似特征确定模块，用于根据所述频域重建图与所述待检测图像之间的差异，确定频域差异掩码，并通过所述频域差异掩码对所述第一图像特征进行自注意力处理，得到频域疑似伪造特征；根据所述色彩重建图与所述待检测图像之间的差异，确定色彩差异掩码，并通过所述色彩差异掩码对所述第二图像特征进行自注意力处理，得到色彩疑似伪造特征；