mmsegmentation/mmseg/models
谢昕辰 5b33faa146
support transformer backbone (#465)
* vit backbone

* fix lint

* add docstrings and fix pretrained pos_embed dim not match prob

* add unittest for vit

* fix lint

* add vit based fcn configs

* fix import error

* support multiple resolution input images

* upsample pos_embed at init_weights

* support resize pos_embed at evaluation

* fix training errors

* add more unitest code for vit backbone

* unitest for uncovered code

* add norm_eval unittest

* refactor _pos_embeding

* minor change

* change var name

* rafactor init_weight

* load weights after resize

* ignore 'module' in pretrain checkpoint

* add with_cp

* add with_cp

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-04-21 20:19:55 -07:00
..
backbones support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
decode_heads Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
losses [Improvement] Add more CI (#460) 2021-04-08 19:29:00 -07:00
necks Add Semantic FPN (#94) 2020-09-03 19:56:36 +08:00
segmentors add BaseSegmentor import to segmentors/__init__.py (#495) 2021-04-19 08:52:42 -07:00
utils refactor test organization (#440) 2021-03-30 17:55:09 -07:00
__init__.py Add Semantic FPN (#94) 2020-09-03 19:56:36 +08:00
builder.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00