模型训练方法和装置、色调映射方法和装置、电子设备、计算机存储介质、以及计算机程序产品

本发明实施例提供了一种模型训练方法和装置、色调映射方法和装置、电子设备、计算机存储介质、以及计算机程序产品。模型训练方法包括：获取训练数据，训练数据包括色调图像样本和色调图像样本的多尺度视觉结构数据；确定包括预训练模块、预配置解析模块和编码模块的扩散模型；以色调图像样本作为扩散模型的主输入和输出，以多尺度视觉结构数据作为预配置解析模块的条件输入，至少对编码模块的网络参数进行迭代训练，得到色调映射模型。预配置解析模块对色调图像样本的上下文特征图进行解析并以多尺度视觉结构数据作为重构条件，对解析结果进行视觉结构特征重构以获得重构特征图，编码模块对重构特征图进行潜在空间编码获得色调图像样本。