一种单图像反射去除方法和装置

本发明公开了一种单图像反射去除方法和装置，涉及图像处理技术领域，包括：采用合成图集对初始反射去除模型进行模型训练验证后，基于自然图集进行模型微调确定目标反射去除模型；将待处理图分别作为透射流初始图与反射流初始图输入粗分离器进行粗分离，对应输出透射粗预测图和反射粗预测图；通过特征编码器分别基于透射粗预测图和反射粗预测图对应确定透射流特征与反射流特征；采用双流交互器根据透射流特征与反射流特征进行流内流间特征增强交互，对应生成改进透射流特征与改进反射流特征；将改进透射流特征和改进反射流特征分别输入上采样层进行上采样，对应输出透射预测图和反射预测图。基于设计的反射去除模型有助于提升图像反射去除可靠性。