zwyang6 / UniA

[Under Review] Tackling Ambiguity from Perspective of Uncertainty Inference and Affinity Diversification for Weakly Supervised Semantic Segmentation
6 stars 0 forks source link
affinity-propagation uncertainty-quantification weakly-supervised-segmentation

Tackling Ambiguity from Perspective of Uncertainty Inference and Affinity Diversification for Weakly Supervised Semantic Segmentation arXiv

News

Overview

We proposed UniA, an unified single-stage framework, to tackle the ambiguity issue in WSSS.

UniA pipeline

Main Results

Semantic performance on VOC and COCO. Logs are available now. Dataset Backbone Val Test Log Weight
PASCAL VOC ViT-B 74.1 73.6 log weight
MS COCO ViT-B 43.2 - log weight

UniA results

Citation

Please cite our work if you find it helpful to your reseach. :two_hearts:

@article{yang2024tackling,
  title={Tackling Ambiguity from Perspective of Uncertainty Inference and Affinity Diversification for Weakly Supervised Semantic Segmentation},
  author={Yang, Zhiwei and Meng, Yucong and Fu, Kexue and Wang, Shuo and Song, Zhijian},
  journal={arXiv preprint arXiv:2404.08195},
  year={2024}
}