mmsegmentation/tests/test_models/test_backbones
谢昕辰 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
..
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_blocks.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_cgnet.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_fast_scnn.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_mobilenet_v3.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_resnest.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_resnet.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_resnext.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_unet.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_vit.py support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
utils.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00