mmsegmentation/mmseg/models/backbones
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
..
__init__.py [Feature]: Add MAE (#1307) 2022-04-28 00:54:20 +08:00
beit.py [Refactor] Add build_pos_embed and build_layers for BEiT (#1517) 2022-04-27 11:37:03 +08:00
bisenetv1.py [Feature] Support BiSeNetV1 (#851) 2021-09-29 02:12:57 +08:00
bisenetv2.py [Enhancement] Add codespell pre-commit hook and fix typos (#920) 2021-10-13 06:21:17 -07:00
cgnet.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
erfnet.py [Feature] Support Real-time model ERFNet (#960) 2021-12-01 23:54:39 -08:00
fast_scnn.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
hrnet.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
icnet.py [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
mae.py [Feature]: Add MAE (#1307) 2022-04-28 00:54:20 +08:00
mit.py [Feature] Add with cp to mit and vit (#1431) 2022-04-01 21:01:45 +08:00
mobilenet_v2.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
mobilenet_v3.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
resnest.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
resnet.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
resnext.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
stdc.py [Feature] Support STDC Network (new) (#995) 2021-12-10 23:09:32 +08:00
swin.py [Fix] Fix type in Swin Transformer (#1274) 2022-02-09 19:08:32 +08:00
timm_backbone.py [Feature] Support TIMMBackbone (#998) 2021-11-02 12:51:11 +08:00
twins.py [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
unet.py [Fix] Fix docstring link problem in readthedocs (#845) 2021-09-02 17:06:43 +08:00
vit.py [Refactor] Refactor transfomer encode layer in ViT and BEiT backbone (#1481) 2022-04-23 17:35:05 +08:00