arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.17469cs.CV
收藏

用于弱监督遥感阴影去除的跨模态球面聚合

Cross-Modal Spherical Aggregation for Weakly Supervised Remote Sensing Shadow Removal

School of Artificial Intelligence, Optics and Electronics (iOPEN), Northwestern Polytechnical University(西北工业大学人工智能与光电学院(iOPEN))

Kaichen Chi, Wei Jing, Junjie Li, Qiang Li, Qi Wang

AI总结:

提出弱监督阴影去除网络S2-ShadowNet,通过球面特征空间融合可见光与红外模态,利用相似性与正交性损失实现特征分离与对齐,并贡献了含4000张图像的大规模基准。

摘要:

遥感阴影去除旨在恢复受污染的地表信息,由于阴影通常表现出极低的照度强度,因此非常棘手。相比之下,红外图像对显著的光照变化具有鲁棒性,提供了与可见光图像互补的视觉线索。然而,现有方法忽略了异构模态之间的协作,导致不希望出现的质量下降。为了填补这一空白,我们提出了一种具有球面特征空间的弱监督阴影去除网络,称为S2-ShadowNet,以探索可见光和红外模态两者的优势。具体而言,我们采用模态转换(可见光到红外)模型来学习跨域映射,从而生成逼真的红外样本。然后,利用Swin Transformer提取具有强表示能力的可见光/红外特征。同时,将提取的特征映射到平滑的球面流形上,通过正则化缓解域偏移。精心设计的相似性损失和正交性损失被嵌入到球面空间中,通过对表示内容和方向的约束,促使私有可见光/红外特征分离以及共享可见光/红外特征对齐。这种方式鼓励了模态之间的隐式互惠,从而为阴影去除提供了新的见解。值得注意的是,实际中无法获得真实标签,因此S2-ShadowNet是通过从阴影图像本身裁剪阴影和无阴影块来进行训练的,避免了刻板且严格的成对数据获取。更重要的是,我们贡献了一个大规模弱监督阴影去除基准,包括4000张带有相应阴影掩膜的阴影图像。

英文摘要:

Remote sensing shadow removal, which aims to recover contaminated surface information, is tricky since shadows typically display overwhelmingly low illumination intensities. In contrast, the infrared image is robust toward significant light changes, providing visual clues complementary to the visible image. Nevertheless, the existing methods ignore the collaboration between heterogeneous modalities, leading to undesired quality degradation. To fill this gap, we propose a weakly supervised shadow removal network with a spherical feature space, dubbed S2-ShadowNet, to explore the best of both worlds for visible and infrared modalities. Specifically, we employ a modal translation (visible-to-infrared) model to learn the cross-domain mapping, thus generating realistic infrared samples. Then, Swin Transformer is utilized to extract strong representational visible/infrared features. Simultaneously, the extracted features are mapped to the smooth spherical manifold, which alleviates the domain shift through regularization. Well-designed similarity loss and orthogonality loss are embedded into the spherical space, prompting the separation of private visible/infrared features and the alignment of shared visible/infrared features through constraints on both representation content and orientation. Such a manner encourages implicit reciprocity between modalities, thus providing a novel insight into shadow removal. Notably, ground truth is not available in practice, thus S2-ShadowNet is trained by cropping shadow and shadow-free patches from the shadow image itself, avoiding stereotypical and strict pair data acquisition. More importantly, we contribute a large-scale weakly supervised shadow removal benchmark, including 4000 shadow images with corresponding shadow masks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑