crystraldo/PSA

This is an official implementation of "Polarized Self-Attention: Towards High-quality Pixel-wise Regression"

Polarized Self-Attention: Towards High-quality Pixel-wise Regression

This is an official implementation of:

Huajun Liu, Fuqiang Liu, Xinyi Fan and Dong Huang. Polarized Self-Attention: Towards High-quality Pixel-wise Regression Arxiv Version

Citation:

@article{Liu2021PSA,
  title={Polarized Self-Attention: Towards High-quality Pixel-wise Regression},
  author={Huajun Liu and Fuqiang Liu and Xinyi Fan and Dong Huang},
  journal={Arxiv Pre-Print arXiv:2107.00782 },
  year={2021}
}

Codes and Pre-trained models will be uploaded soon~

Top-down 2D pose estimation models pre-trained on the MS-COCO keypoint task(Table4 in the Arxiv version).

Model Name	Backbone	Input Size	AP	pth file
UDP-Pose-PSA(p)	HRNet-W48	256x192	78.9	to be uploaded
UDP-Pose-PSA(p)	HRNet-W48	384x288	79.5	to be uploaded
UDP-Pose-PSA(s)	HRNet-W48	384x288	79.4	to be uploaded

Setup and inference:

Semantic segmentation models pre-trained on Cityscapes (Table5 in the Arxiv version).

Model Name	Backbone	val mIoU	pth file
HRNetV2-OCR+PSA(p)	HRNetV2-W48	86.95	download
HRNetV2-OCR+PSA(s)	HRNetV2-W48	86.72	download

Setup and inference: