mmsegmentation/tests
谢昕辰 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
..
data init commit 2020-07-10 02:39:01 +08:00
test_data [Bug Fix] Fix TTA resize scale (#334) 2021-01-07 09:58:34 -08:00
test_models support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_config.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
test_eval_hook.py Add runner type (#118) 2020-10-25 15:18:46 -07:00
test_inference.py Bugfix: num of augmentations and image meta not match when run TTA on CPU (#276) 2020-12-13 23:51:42 -08:00
test_metrics.py Fix mIoU calculatiton range (#471) 2021-04-14 08:37:23 -07:00
test_sampler.py Generalized OHEM (#54) 2020-08-09 23:49:23 +08:00