The code of visual center

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.

https://hangzhang.org/PyTorch-Encoding/

MIT License

2.04k stars 450 forks source link

The code of visual center #409

Open QY1994-0919 opened 2 years ago

QY1994-0919 commented 2 years ago

After reading this paper, I was interested in the proposal of visual center in the paper, but I could not find the corresponding code of the specific part? How to choose the codewords (visual centers) ?

zhanghang1989 commented 2 years ago

If you could read Chinese, you may refer to this link: https://github.com/zhanghang1989/PyTorch-Encoding/issues/384#issuecomment-957086497

The mmsegmentation implementation is easier to read https://github.com/open-mmlab/mmsegmentation/blob/master/mmseg/ops/encoding.py#L6