mmsegmentation/mmseg
Yuan Liu 6563cb513e
[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
..
apis [Fix] Register optimizer constructor with mmseg (#1456) 2022-04-16 16:45:17 +08:00
core [Refactory] Merge BEiT and ConvNext 's LR decay optimizer constructors (#1438) 2022-04-27 10:43:25 +08:00
datasets Avoid converting label ids twice by label map during evaluation (#1417) 2022-04-12 18:26:25 +08:00
models [Feature]: Add MAE (#1307) 2022-04-28 00:54:20 +08:00
ops [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
utils add multi-processes script (#1238) 2022-01-27 21:18:55 +08:00
__init__.py [Enhancement] MMCV maximum_version set to 1.6 (#1480) 2022-04-19 16:19:56 +08:00
version.py Bump version to v0.23.0 (#1439) 2022-04-01 15:43:19 +08:00