mmsegmentation/tests/test_models
Yuan Liu 43b8a1d8dc [Feature]: Add MAE (#1307)
* [Fix]: Fix lint

* [WIP]: Add mae seg config

* [Feature]: Add MAE seg

* [Fix]: Fix mae dataset img scale bug

* [Fix]: Fix lint

* [Feature]: Change mae config to mae_segmentation's config

* [Feature]: Add interpolate pe when loading

* [Fix]: Fix pos_embed not used bug

* [Fix]: Fix lint

* [Fix]: Init rel pos embed with zeros

* [Fix]: Fix lint

* [Fix]: Change the type name of backbone to MAE

* [Fix]: Delete ade20k_512x512.py

* [Fix]: Use mmseg provided ade20k.py

* [Fix]: Change 1 sample per gpu to 2 samples per gpu

* [Fix]: Fix conflict

* [Refactor]: Use the TransformerEncoderLayer of  BEiT

* [Feature]: Add UT

* [Fix]: Change the default value of qv bias to False

* [Fix]: Initialize relative pos table with zeros

* [Fix]: Delete redundant code in mae

* [Fix]: Fix lint

* [Fix]: Rename qkv_bias to qv_bias

* [Fix]: Add docstring to weight_init of MAEAttention

* [Refactor]: Delete qv_bias param

* [Fix]: Add reference to fix_init_weight

* [Fix]: Fix lint

* [Fix]: Delete extra crop size

* [Refactor]: Rename mae

* [Fix]: Set bias to True

* [Fix]: Delete redundant params

* [Fix]: Fix lint

* [Fix]: Fix UT

* [Fix]: Add resize abs pos embed

* [Fix]: Fix UT

* [Refactor]: Use build layer

* [Fix]: Add licsense and fix docstring

* [Fix]: Fix docstring

* [Feature]: Add README metafile

* [Fix]: Change 640 to 512

* [Fix]: Fix README

* fix readme of MAE

Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
2022-04-28 00:54:20 +08:00
..
test_backbones [Feature]: Add MAE (#1307) 2022-04-28 00:54:20 +08:00
test_heads [Feature] Support K-Net (#1289) 2022-03-10 20:05:37 +08:00
test_losses [Fix] Support single cahnnel `pred` for Binary Cross Entropy Loss (#1454) 2022-04-14 11:26:02 +08:00
test_necks [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
test_segmentors [Fix] Fix loss parse in val_step (#906) 2021-09-25 18:17:40 -07:00
test_utils [Feature] add nlc2nchw2nlc and nchw2nlc2nchw (#1249) 2022-03-10 20:27:28 +08:00
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_forward.py [Fix] Convert SyncBN to BN when training on DP (#772) 2021-09-15 09:39:37 -07:00