[发明专利]图像增强模型的训练方法和对图像进行增强的方法

说明书

本公开提供了一种图像增强模型的训练方法和对图像进行增强的方法、装置、设备和存储介质，涉及人工智能技术领域，具体涉及计算机视觉和深度学习技术领域，可应用于图像处理场景下。图像增强模型的训练方法包括：针对当前训练轮次，以样本图像分别作为教师网络模型和学生网络模型的输入，分别获得第一增强图像和第二增强图像；基于第一增强图像和第二增强图像对学生网络模型进行训练，获得针对当前训练轮次的训练后学生网络模型；以及基于针对当前轮次的训练后学生网络模型参数的第一取值和前一训练轮次确定的图像增强模型参数的第二取值，确定当前训练轮次中图像增强模型参数的取值。其中，图像增强模型的初始模型与学生网络模型的初始模型相同。

技术领域

本公开涉及人工智能技术领域，具体涉及计算机视觉和深度学习技术领域，可应用于图像处理场景下。

背景技术

随着电子技术的发展，将模糊图像变为清晰图像的技术在安防、摄像等领域得到了广泛的应用。该将模糊图像变为清晰图像的技术一般对电子设备的处理效率要求很高，需要在云端部署很大的算力。由于高处理效率的要求，通常难以实现对图像的实时处理，且无法在终端设备等处理效率低的设备上部署该技术。

发明内容

提供了一种便于降低算力要求的图像增强模型的训练方法和对图像进行增强的方法、装置、设备和存储介质。

根据本公开的一个方面，提供了一种图像增强模型的训练方法，包括：针对当前训练轮次：以样本图像分别作为教师网络模型和学生网络模型的输入，分别获得第一增强图像和第二增强图像；基于第一增强图像和第二增强图像，对学生网络模型进行训练，获得针对当前训练轮次的训练后学生网络模型；以及基于针对当前训练轮次的训练后学生网络模型的参数的第一取值和前一训练轮次确定的图像增强模型的参数的第二取值，确定当前训练轮次中图像增强模型的参数的取值，其中，图像增强模型的初始模型与学生网络模型的初始模型相同。

根据本公开的另一个方面，提供了一种对图像进行增强的方法，包括：将待增强图像输入图像增强模型；以及采用图像增强模型对待增强图像进行处理，获得待增强图像的第三增强图像，其中，图像增强模型是上述的图像增强模型的训练方法训练的。

根据本公开的另一个方面，提供了一种图像增强模型的训练装置，包括：增强图像获得模块，用于针对当前训练轮次，以样本图像分别作为教师网络模型和学生网络模型的输入，分别获得第一增强图像和第二增强图像；模型训练模块，用于基于第一增强图像和第二增强图像，对学生网络模型进行训练，获得针对当前训练轮次的训练后学生网络模型；以及参数取值确定模块，用于基于针对当前训练轮次的训练后学生网络模型的参数的第一取值和前一训练轮次确定的图像增强模型的参数的第二取值，确定当前训练轮次中图像增强模型的参数的取值。其中，图像增强模型的初始模型与学生网络模型的初始模型相同。

根据本公开的另一个方面，提供了一种对图像进行增强的装置，包括：图像输入模块，用于将待增强图像输入图像增强模型；以及图像增强模块，用于采用图像增强模型对待增强图像进行处理，获得待增强图像的第三增强图像，其中，图像增强模型是采用上述的图像增强模型的训练装置训练的。

根据本公开的另一个方面，提供了一种电子设备，包括：至少一个处理器；以及与至少一个处理器通信连接的存储器；其中，存储器存储有可被至少一个处理器执行的指令，指令被至少一个处理器执行，以使至少一个处理器能够执行本公开提供的图像增强模型的训练方法和对图像进行增强的方法。

根据本公开的另一个方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，计算机指令用于使计算机执行本公开提供的图像增强模型的训练方法和对图像进行增强的方法。

根据本公开的另一个方面，提供了一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现本公开提供的图像增强模型的训练方法和对图像进行增强的方法。