[发明专利]图像处理方法、电子装置及芯片

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

使用暗光图像数据集和增强图像数据集对初始增强模型进行训练，获得图像增强模型；其中，所述初始增强模型用于基于解耦表征提取至少一种图像特征，并根据所述至少一种图像特征进行图像增强处理；

通过所述图像增强模型对待处理暗光图像进行增强处理，获得增强后图像。

2.根据权利要求1所述的方法，其特征在于，

所述初始增强模型包括内容编码器，暗光编码器，内容图像生成器，暗光图像生成器；

所述图像增强模型包括训练后的内容编码器和训练后的内容图像生成器；其中，

所述内容编码器，用于提取输入图像的内容特征，并将所述内容特征分别输出至所述内容图像生成器和所述暗光图像生成器；

所述暗光编码器，用于提取输入图像的暗光特征，并将所述暗光特征输出至所述暗光图像生成器；

所述内容图像生成器，用于根据输入的所述内容特征进行图像生成处理；

所述暗光图像生成器，用于根据输入的所述内容特征和所述暗光特征进行图像重建处理。

3.根据权利要求2所述的方法，其特征在于，所述使用暗光图像数据集和增强图像数据集对初始增强模型进行训练，获得图像增强模型，包括：

通过所述内容编码器和所述暗光编码器，分别获取所述暗光图像数据集中的第一图像对应的第一内容特征和第一暗光特征，以及所述增强图像数据集中的第二图像对应的第二内容特征；

基于所述第一内容特征、所述第二内容特征以及所述第一暗光特征，通过所述内容图像生成器和所述暗光图像生成器进行生成处理，分别获得第一增强图像、第一重建图像、第二重建图像以及第一暗光图像；

通过所述内容编码器和所述暗光编码器，分别获取所述第一增强图像对应的第三内容特征，以及所述第一暗光图像对应的第四内容特征和第二暗光特征；

基于所述第三内容特征、所述第四内容特征以及所述第二暗光特征，通过所述内容图像生成器和所述暗光图像生成器进行生成处理，分别获得第一循环图像和第二循环图像；

使用所述第一重建图像，所述第二重建图像，所述第一增强图像，所述第一暗光图像，所述第一循环图像以及所述第二循环图像，对所述内容编码器和所述内容图像生成器进行无监督训练，获得所述训练后的内容编码器和所述训练后的内容图像生成器。

4.根据权利要求3所述的方法，其特征在于，所述通过所述内容编码器和所述暗光编码器，分别获取所述暗光图像数据集中的第一图像对应的第一内容特征和第一暗光特征，以及所述增强图像数据集中的第二图像对应的第二内容特征，包括：

通过所述内容编码器提取所述第一图像对应的所述第一内容特征；

通过所述暗光编码器提取所述第一图像对应的所述第一暗光特征；

通过所述内容编码器提取所述第二图像对应的所述第二内容特征。

5.根据权利要求3所述的方法，其特征在于，所述基于所述第一内容特征、所述第二内容特征以及所述第一暗光特征，通过所述内容图像生成器和所述暗光图像生成器进行生成处理，分别获得第一增强图像、第一重建图像、第二重建图像以及第一暗光图像，包括：

将所述第一内容特征输入至所述内容图像生成器，获得所述第一图像对应的所述第一增强图像；

将所述第一内容特征和所述第一暗光特征输入至所述暗光图像生成器，获得所述第一图像对应的所述第一重建图像；

将所述第二内容特征输入至所述内容图像生成器，获得所述第二图像对应的所述第二重建图像；

将所述第二内容特征和所述第一暗光特征输入至所述暗光图像生成器，获得所述第二图像对应的所述第一暗光图像。

6.根据权利要求3所述的方法，其特征在于，所述通过所述内容编码器和所述暗光编码器，分别获取所述第一增强图像对应的第三内容特征，以及所述第一暗光图像对应的第四内容特征和第二暗光特征，包括：

通过所述内容编码器提取所述第一增强图像对应的所述第三内容特征；

通过所述内容编码器提取所述第一暗光图像对应的所述第四内容特征；

通过所述暗光编码器提取所述第一暗光图像对应的所述第二暗光特征。