mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/configs/_base_/models
History
Ze Liu b6c7c77a08
[WIP] Add Swin Transformer (#511) 
* add Swin Transformer

* add Swin Transformer

* fixed import

* Add some swin training settings.

* Fix some filename error.

* Fix attribute name: pretrain -> pretrained

* Upload mmcls implementation of swin transformer.

* Refactor Swin Transformer to follow mmcls style.

* Refactor init_weigths of swin_transformer.py

* Fix lint

* Match inference precision

* Add some comments

* Add swin_convert to load official style ckpt

* Remove arg: auto_pad

* 1. Complete comments for each block;

2. Correct weight convert function;

3. Fix the pad of Patch Merging;

* Clean function args.

* Fix vit unit test.

* 1. Add swin transformer unit tests;

2. Fix some pad bug;

3. Modify config to adapt new swin implementation;

* Modify config arg

* Update readme.md of swin

* Fix config arg error and Add some swin benchmark msg.

* Add MeM and ms test content for readme.md of swin transformer.

* Fix doc string of swin module

* 1. Register swin transformer to model list;

2. Modify pth url which keep meta attribute;

* Update swin.py

* Merge config settings.

* Modify config style.

* Update README.md

Add ViT link

* Modify main readme.md

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
Co-authored-by: sennnnn <201730271412@mail.scut.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
 		2021-07-01 23:41:55 +08:00
..
ann_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
apcnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
ccnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
cgnet.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
danet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
deeplabv3_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
deeplabv3_unet_s5-d16.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
deeplabv3plus_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
dmnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
dnl_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
emanet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
encnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
fast_scnn.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
fcn_hr18.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
fcn_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
fcn_unet_s5-d16.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
fpn_r50.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
gcnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
lraspp_m-v3-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
nonlocal_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
ocrnet_hr18.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
ocrnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
pointrend_r50.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
psanet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
pspnet_r50-d8.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
pspnet_unet_s5-d16.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
setr_mla.py [Feature] Official implementation of SETR (#531) 2021-06-23 09:39:29 -07:00
setr_naive.py [Feature] Official implementation of SETR (#531) 2021-06-23 09:39:29 -07:00
setr_pup.py [Feature] Official implementation of SETR (#531) 2021-06-23 09:39:29 -07:00
upernet_r50.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
upernet_swin.py [WIP] Add Swin Transformer (#511) 2021-07-01 23:41:55 +08:00
upernet_vit-b16_ln_mln.py add configs for vit backbone plus decode_heads (#520) 2021-07-01 23:00:39 +08:00