一种去除图像反光模型的训练方法及装置

本发明公开了一种去除图像反光模型的训练方法及装置，包括获取的对待去反光训练原图进行预处理，输出待训练初始去反光特征图和待训练初始反光特征图；通过初始传输预测网络对待训练初始去反光特征图进行传输预测，输出当前时刻待训练目标去反光特征图和传输解码特征图；采用初始反射预测网络对待训练初始反光特征图进行反射预测，生成当前时刻待训练目标反光特征图；根据当前时刻待训练目标去反光特征图、待去反光训练原图、当前时刻待训练目标反光特征图和传输解码特征图对初始去除图像反光模型进行模型训练，确定目标去除图像反光模型；解决了现有的去除图像反光技术导致图像进行反光去除后图像的质量降低的技术问题。