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

ShareCMP:偏振感知的RGB-P语义分割

ShareCMP: Polarization-Aware RGB-P Semantic Segmentation

Harbin Engineering University(哈尔滨工程大学)

Zhuoyan Liu, Bo Wang, Lizhi Wang, Chenyu Mao, Ye Li

AI总结:

本文提出ShareCMP,一种共享双分支的RGB-P语义分割框架,通过PGA模块和CPALoss/CPAAHead增强偏振信息利用,在三个基准上以更少参数取得最佳mIoU。

摘要:

多模态语义分割正在快速发展,但RGB-偏振(RGB-P)模态仍未得到充分探索。为深入研究这一问题,我们构建了一个包含12个典型水下语义类别的UPLight RGB-P分割基准。在本工作中,我们设计了ShareCMP,一种具有共享双分支架构(ShareCMP Encoder)的RGB-P语义分割框架,与以往的双分支模型相比,其参数量和内存空间减少了约33.8%。该框架包含一个偏振生成注意力(PGA)模块,旨在为编码器生成具有更丰富偏振特性的偏振模态图像。此外,我们引入了类别偏振感知损失(CPALoss)和类别偏振感知辅助头(CPAAHead),以增强编码器对偏振模态信息的学习与理解,并优化PGA模块。通过在三个RGB-P基准上的大量实验,我们的ShareCMP在UPLight(92.45(+0.32)%)、ZJU(92.7(+0.1)%)和MCubeS(50.99(+1.51)%)数据集上以更少的参数取得了最佳的mIoU性能。并且,与其他RGB-X数据集上相应的最先进RGB-X方法相比,我们的ShareCMP(不含PGA)取得了具有竞争力甚至更高的性能。代码和数据集可在https://github.com/LEFTeyex/ShareCMP获取。

英文摘要:

Multimodal semantic segmentation is developing rapidly, but the modality of RGB-\textbf{P}olarization remains underexplored. To delve into this problem, we construct a UPLight RGB-P segmentation benchmark with 12 typical underwater semantic classes. In this work, we design the ShareCMP, an RGB-P semantic segmentation framework with a shared dual-branch architecture (ShareCMP Encoder), which reduces the parameters and memory space by about 33.8\% compared to previous dual-branch models. It encompasses a Polarization Generate Attention (PGA) module designed to generate polarization modal images with richer polarization properties for the encoder. In addition, we introduce the Class Polarization-Aware Loss (CPALoss) with Class Polarization-Aware Auxiliary Head (CPAAHead) to improve the learning and understanding of the encoder for polarization modal information and to optimize the PGA module. With extensive experiments on a total of three RGB-P benchmarks, our ShareCMP achieves the best performance in mIoU with fewer parameters on the UPLight (92.45{\small (+0.32)}\%), ZJU (92.7{\small (+0.1)}\%), and MCubeS (50.99{\small (+1.51)}\%) datasets. And our ShareCMP (w/o PGA) achieves competitive or even higher performance on other RGB-X datasets compared to the corresponding state-of-the-art RGB-X methods. The code and datasets are available at https://github.com/LEFTeyex/ShareCMP.

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

↑