CAA: Class-Aware Affinity calculation add-on for semantic segmentation

Huadong Tang, Youpeng Zhao, Chaofan Du, Min Xu, Qiang Wu

Abstract: Highlights•A class-aware affinity module is proposed.•Affinity map is proposed to study both pixel-to-pixel relations.•Class center is proposed to explore the pixel-to-class associations.•Our module can be incorporated into existing segmentation frameworks.