首页|期刊导航|计算机技术与发展|基于掩码引导融合与双卷积前馈的去阴影网络

基于掩码引导融合与双卷积前馈的去阴影网络OA

Shadow Removal Network Based on Mask-guided Fusion and Dual Convolution Feedforward

中文摘要英文摘要

阴影作为图像中的常见现象,常会影响图像质量及后续视觉任务的准确性,尤其是在复杂背景和不同光照条件下.现有的阴影去除方法在处理这些挑战时,常面临边缘残留、颜色不一致和纹理丢失等问题,严重制约了其在实际应用中的有效性和普适性.为此,该文提出一种基于 Transformer 架构的阴影去除网络—掩码引导融合与前馈网络(Mask-Guided Fusion and Feedforward Network,MGFF-Net).该网络包含两个关键模块:掩码引导融合模块(MGFN)通过引入显式阴影掩码信息并结合门控机制,提升对阴影区域的建模能力,显著减少了阴影区域的色差与边缘模糊;双卷积前馈模块(DualConvFFN)则融合局部纹理与全局语义特征,增强细节还原与结构保持能力.在 SRD、ISTD 和 ISTD+三个公开数据集上的实验表明,MGFF-Net 在 PSNR、SSIM 和 RMSE 等多个评价指标上展现出较为优越的性能.具体而言,在 SRD 数据集上,与主流方法相比,整体图像的PSNR 提升了1.35 dB,RMSE 降低了17.3%,SSIM 基本持平;在ISTD+数据集上,整体图像的 PSNR 提升了0.95 dB,RMSE 降低了11.03%,SSIM 基本持平.实验结果验证了该方法在细节恢复、结构保持和阴影区域恢复方面的优势.

Shadows are a common phenomenon in images that often degrade image quality and impair the accuracy of subsequent visual tasks,especially under complex backgrounds and varying lighting conditions.Existing shadow removal methods frequently encounter challenges such as edge residues,color inconsistencies,and texture loss,which significantly hinder their effectiveness and generalizability in practical applications.To address these issues,we propose a Transformer-based shadow removal network—Mask-Guided Fusion and Feedforward Network(MGFF-Net).The network consists of two key modules:the Mask-Guided Fusion Module(MGFM),which in-corporates explicit shadow mask information and employs a gating mechanism to enhance the model's ability to capture shadow regions,significantly reducing color discrepancies and edge blurring in the shadow areas;and the Dual Convolution Feedforward Module(Dual-ConvFFN),which integrates local texture and global semantic features to improve detail restoration and structural preservation.Experiments conducted on three public datasets—SRD,ISTD,and ISTD+—demonstrate that MGFF-Net achieves superior performance across several evaluation metrics,including PSNR,SSIM,and RMSE.Specifically,on the SRD dataset,compared to mainstream methods,the overall image PSNR increases by 1.35 dB,RMSE decreases by 17.3%,and SSIM remains roughly the same;on the ISTD+dataset,the overall image PSNR improves by0.95 dB,and RMSE decreases by11.03%,and SSIM remains roughly the same.The ex-perimental results validate the advantages of the proposed method in detail recovery,structural consistency,and shadow region restoration.

杜启鲁;张凡龙

南京审计大学 计算机学院,江苏 南京 211815南京审计大学 计算机学院,江苏 南京 211815

信息技术与安全科学

阴影去除Transformer门控机制掩码引导融合模块双卷积前馈模块

shadow removalTransformergating mechanismmask-guided fusion moduledual convolution feedforward module

《计算机技术与发展》 2026 (4)

47-54,8

国家自然科学基金(62276137)

10.20165/j.cnki.ISSN1673-629X.2025.0302

评论