mmsegmentation/mmseg
谢昕辰 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
..
apis add dynamic export and visualize to pytorch2onnx (#463) 2021-04-12 11:54:59 -07:00
core Fix mIoU calculatiton range (#471) 2021-04-14 08:37:23 -07:00
datasets Add support for Pascal Context 59 classes (#459) 2021-04-19 21:20:54 -07:00
models support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
ops add dynamic export and visualize to pytorch2onnx (#463) 2021-04-12 11:54:59 -07:00
utils [Enhance] Refactor inverted residual (#164) 2020-09-28 00:33:51 +08:00
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
version.py Bump to v0.12 (#455) 2021-04-03 21:02:56 -07:00