Home

Awesome

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

PWC PWC PWC

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 NameBackboneInput SizeAPpth file
UDP-Pose-PSA(p)HRNet-W48256x19278.9to be uploaded
UDP-Pose-PSA(p)HRNet-W48384x28879.5to be uploaded
UDP-Pose-PSA(s)HRNet-W48384x28879.4to be uploaded

Setup and inference:

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

Model NameBackboneval mIoUpth file
HRNetV2-OCR+PSA(p)HRNetV2-W4886.95download
HRNetV2-OCR+PSA(s)HRNetV2-W4886.72download

Setup and inference: