Positive-negative equal contrastive loss for semantic segmentation

This repository is the PyTorch Implementation.

Our experiment is based on the MMsegmentation 0.13 version.

Due to our proposed PNE loss being plug-on-play, we provide the code of PNE loss and an example to apply the PNE loss on the ASPP decoder in the MMsegmentation framework.

Citing PNE loss

@article{wang2023positive,
  title={Positive-negative equal contrastive loss for semantic segmentation},
  author={Wang, Jing and Li, Jiangyun and Li, Wei and Xuan, Lingfei and Zhang, Tianxiang and Wang, Wenxuan},
  journal={Neurocomputing},
  year={2023},
  publisher={Elsevier}
}