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
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 |