一种基于条件扩散模型的两阶段雨雾图像修复方法

本发明公开了一种基于条件扩散模型的两阶段雨雾图像修复方法，属于数字图像处理技术领域，包括以下步骤：步骤1：获取待修复的雨雾图像数据并对其进行预处理；步骤2：将处理后的图像输入到训练好的退化感知模型进行处理，移除与图像背景内容无关的天气退化前景，获得第一阶段的输出X；步骤3：将步骤2输出X输入语义信息提取器，获取语义特征F；步骤4：将步骤2输出X和步骤3的输出F同时送入语义感知条件扩散模型，得到最终的修复图像。本发明通过将雨雾图像修复任务分解为退化移除和内容重建两个子任务，能够在移除恶劣天气前景的基础上，重建出与原图在语义上更加一致的纹理细节，实现良好的图像修复效果。