-
- Downloads
[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>
Showing
- configs/_base_/models/upernet_mae.py 49 additions, 0 deletionsconfigs/_base_/models/upernet_mae.py
- configs/mae/README.md 81 additions, 0 deletionsconfigs/mae/README.md
- configs/mae/mae.yml 23 additions, 0 deletionsconfigs/mae/mae.yml
- configs/mae/upernet_mae-base_fp16_512x512_160k_ade20k_ms.py 24 additions, 0 deletionsconfigs/mae/upernet_mae-base_fp16_512x512_160k_ade20k_ms.py
- configs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py 48 additions, 0 deletionsconfigs/mae/upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py
- mmseg/models/backbones/__init__.py 2 additions, 1 deletionmmseg/models/backbones/__init__.py
- mmseg/models/backbones/mae.py 261 additions, 0 deletionsmmseg/models/backbones/mae.py
- model-index.yml 1 addition, 0 deletionsmodel-index.yml
- tests/test_models/test_backbones/test_mae.py 183 additions, 0 deletionstests/test_models/test_backbones/test_mae.py
Loading
Please register or sign in to comment