mmsegmentation/mmseg/models/backbones
sennnnn cf2cb542f7
Adjust vision transformer backbone architectures (#524)
* Adjust vision transformer backbone architectures;

* Add DropPath, trunc_normal_ for VisionTransformer implementation;

* Add class token buring intermediate period and remove it during final period;

* Fix some parameters loss bug;

* * Store intermediate token features and impose no processes on them;

* Remove class token and reshape entire token feature from NLC to NCHW;

* Fix some doc error

* Add a arg for VisionTransformer backbone to control if input class token into transformer;

* Add stochastic depth decay rule for DropPath;

* * Fix output bug when input_cls_token=False;

* Add related unit test;

* * Add arg: out_indices to control model output;

* Add unit test for DropPath;

* Apply suggestions from code review

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-04-30 10:37:47 -07:00
..
__init__.py support transformer backbone (#465) 2021-04-21 20:19:55 -07:00
cgnet.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
fast_scnn.py [Enhance] Refactor inverted residual (#164) 2020-09-28 00:33:51 +08:00
hrnet.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
mobilenet_v2.py [Enhance] Refactor inverted residual (#164) 2020-09-28 00:33:51 +08:00
mobilenet_v3.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
resnest.py Support ResNeSt backbone (#47) 2020-08-17 00:54:01 +08:00
resnet.py Add Colab Tutorial (#7) 2020-07-10 16:55:47 +08:00
resnext.py init commit 2020-07-10 02:39:01 +08:00
unet.py Save base_channels for UNet (#450) 2021-03-30 09:52:08 -07:00
vit.py Adjust vision transformer backbone architectures (#524) 2021-04-30 10:37:47 -07:00