一种基于扩散模型的文档图像光照恢复方法及系统

一种基于扩散模型的文档图像光照恢复方法及系统，恢复方法包括以下步骤：构建一个条件扩散模块，该模块以阴影图像作为条件输入，同时输入噪声图像，预测条件评分函数；设计一个文档图像光照恢复模块，采用由粗到细的处理策略，通过粗网络预测初步的退化图，再结合阴影图像、评分函数和初步退化映射，通过精网络预测更精确的退化映射；将精确的退化映射应用于阴影图像，实现文档图像的光照恢复；构建一个高分辨率、具有丰富布局的复杂照明恢复测试集，评估模型的泛化能力。本发明实现了对文档图像光照的有效恢复，同时提高了处理高分辨率图像的效率，解决了现有技术在恢复文档图像光照时存在的恢复效果不佳、处理效率低及生成结果不一致的问题。