mmsegmentation/tests
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
..
data init commit 2020-07-10 02:39:01 +08:00
test_data add metric mFscore (#509) 2021-04-30 11:34:57 -07:00
test_models [WIP] Add Swin Transformer (#511) 2021-07-01 23:41:55 +08: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 Bump to v0.13 (#529) 2021-05-05 16:56:19 -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 add metric mFscore (#509) 2021-04-30 11:34:57 -07:00
test_sampler.py Generalized OHEM (#54) 2020-08-09 23:49:23 +08:00