一种用于图像光影和谐的AI协调方法

本发明公开了一种用于图像光影和谐的AI协调方法，属于图像协调技术领域，包括采用实拍合成数据和路径追踪渲染数据的混合方式构建输入图像的数据集；使用前沿矫正流变换器扩散模型作为基础架构，将不协调的输入图像编码为潜在噪声向量和光照条件向量；基于特征提取层的参数，对输入图像进行图像特征提取，得到图像特征向量；同时提取全景光照特征向量；将图像特征向量和全景光照特征向量输入融合层，进行多模态注意力融合，得到融合后的特征向量；将融合后的特征向量输入生成层，根据速度预测器迭代refinement，得到生成图像。本发明采用上述方法，实现光线、阴影、投影及反射的全方位重计算，让前景与背景的光影交互符合物理规律。