光照主方向模型的训练方法及电子设备

本申请提供一种光照主方向模型的训练方法及电子设备，涉及图像处理技术领域。该方法中，通过对原始图像进行重打光数据增强得到样本图像对(正样本图像对和/或负样本图像对)；基于样本图像对无监督地训练第一编码器，得到第一编码器的权重参数；将第一编码器的权重参数迁移至待训练的光照主方向估计模型，基于携带标签的样本图像有监督地训练待训练的光照主方向估计模型。如此，可以基于无监督训练得到的第一编码器的权重参数对光照主方向估计模型进行训练，从而降低训练数据集的获取难度。