基于全局上下文感知的单图像阴影去除方法及计算机装置

本发明涉及阴影去除领域，具体涉及一种基于全局上下文感知的单图像阴影去除方法及计算机装置。方案包括：对原始数据集进行预处理，然后引入基于Retinex理论的阴影模型，该模型能够描述图像中阴影区域和非阴影区域之间的光照和颜色差异；使用多尺度通道注意力机制构建的编码器和解码器，高效捕捉层次化的全局信息；基于编码器和解码器构建基于上下文感知的阴影去除网络，基于阴影模型，在阴影去除网络的瓶颈阶段加入阴影交互模块和阴影交互注意力机制；使用损失函数训练阴影去除网络；利用公共数据集对训练好的阴影去除网络进行测试以评估阴影去除网络的性能；最后根据评估后的阴影去除网络进行图像阴影去除。本发明适用于单图像阴影去除。