mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/mmseg/models/decode_heads
History
sennnnn b4fd32d049 [Feature] Add segformer decode head and related train config (#599) 
* [Feature]Segformer re-implementation

* Using act_cfg and norm_cfg to control activation and normalization

* Split this PR into several little PRs

* Fix lint error

* Remove SegFormerHead

* [Feature] Add segformer decode head and related train config

* Add ade20K trainval support for segformer

1. Add related train and val configs;

2. Add AlignedResize;

* Set arg: find_unused_parameters = True

* parameters init refactor

* 1. Refactor segformer backbone parameters init;

2. Remove rebundant functions and unit tests;

* Remove rebundant codes

* Replace Linear Layer to 1X1 Conv

* Use nn.ModuleList to refactor segformer head.

* Remove local to_xtuple

* 1. Remove rebundant codes;

2. Modify module name;

* Refactor the backbone of segformer using mmcv.cnn.bricks.transformer.py

* Fix some code logic bugs.

* Add mit_convert.py to match pretrain keys of segformer.

* Resolve some comments.

* 1. Add some assert to ensure right params;

2. Support flexible peconv position;

* Add pe_index assert and fix unit test.

* 1. Add doc string for MixVisionTransformer;

2. Add some unit tests for MixVisionTransformer;

* Use hw_shape to pass shape of feature map.

* 1. Fix doc string of MixVisionTransformer;

2. Simplify MixFFN;

3. Modify H, W to hw_shape;

* Add more unit tests.

* Add doc string for shape convertion functions.

* Add some unit tests to improve code coverage.

* Fix Segformer backbone pretrain weights match bug.

* Modify configs of segformer.

* resolve the shape convertion functions doc string.

* Add pad_to_patch_size arg.

* Support progressive test with fewer memory cost.

* Modify default value of pad_to_patch_size arg.

* Temp code

* Using processor to refactor evaluation workflow.

* refactor eval hook.

* Fix process bar.

* Fix middle save argument.

* Modify some variable name of dataset evaluate api.

* Modify some viriable name of eval hook.

* Fix some priority bugs of eval hook.

* Fix some bugs about model loading and eval hook.

* Add ade20k 640x640 dataset.

* Fix related segformer configs.

* Depreciated efficient_test.

* Fix training progress blocked by eval hook.

* Depreciated old test api.

* Modify error patch size.

* Fix pretrain of mit_b0

* Fix the test api error.

* Modify dataset base config.

* Fix test api error.

* Modify outer api.

* Build a sampler test api.

* TODO: Refactor format_results.

* Modify variable names.

* Fix num_classes bug.

* Fix sampler index bug.

* Fix grammaly bug.

* Add part of benchmark results.

* Support batch sampler.

* More readable test api.

* Remove some command arg and fix eval hook bug.

* Support format-only arg.

* Modify format_results of datasets.

* Modify tool which use test apis.

* Update readme.

* Update readme of segformer.

* Updata readme of segformer.

* Update segformer readme and fix segformer mit_b4.

* Update readme of segformer.

* Clean AlignedResize related config.

* Clean code from pr #709

* Clean code from pr #709

* Add 512x512 segformer_mit-b5.

* Fix lint.

* Fix some segformer head bugs.

* Add segformer unit tests.

* Replace AlignedResize to ResizeToMultiple.

* Modify readme of segformer.

* Fix bug of ResizeToMultiple.

* Add ResizeToMultiple unit tests.

* Resolve conflict.

* Simplify the implementation of ResizeToMultiple.

* Update test results.

* Fix multi-scale test error when resize_ratio=1.75 and input size=640x640.

* Update segformer results.

* Update Segformer results.

* Fix some url bugs and pipelines bug.

* Move ckpt convertion to tools.

* Add segformer official pretrain weights usage.

* Clean redundant codes.

* Remove redundant codes.

* Unfied format.

* Add description for segformer converter.

* Update workers.
 		2021-08-13 13:31:19 +08:00
..
__init__.py [Feature] Add segformer decode head and related train config (#599) 2021-08-13 13:31:19 +08:00
ann_head.py init commit 2020-07-10 02:39:01 +08:00
apc_head.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
aspp_head.py init commit 2020-07-10 02:39:01 +08:00
cascade_decode_head.py init commit 2020-07-10 02:39:01 +08:00
cc_head.py init commit 2020-07-10 02:39:01 +08:00
da_head.py init commit 2020-07-10 02:39:01 +08:00
decode_head.py [Refactor]: Unified parameter initialization (#567) 2021-06-16 21:41:29 -07:00
dm_head.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
dnl_head.py Add "disentangled non-local (DNL) neural networks" [ECCV2020] (#37) 2020-09-07 16:22:00 +08:00
ema_head.py [Feature] Support EMANet (#34) 2020-09-07 13:06:59 +08:00
enc_head.py init commit 2020-07-10 02:39:01 +08:00
fcn_head.py Support fcn dilate 6 (#400) 2021-03-20 10:32:04 -07:00
fpn_head.py [Fix] Replace interpolate with resize (#731) 2021-07-28 16:56:22 +08:00
gc_head.py Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
lraspp_head.py [New model] Support MobileNetV3 (#268) 2020-12-26 00:02:50 -08:00
nl_head.py init commit 2020-07-10 02:39:01 +08:00
ocr_head.py init commit 2020-07-10 02:39:01 +08:00
point_head.py [Refactor]: Unified parameter initialization (#567) 2021-06-16 21:41:29 -07:00
psa_head.py init commit 2020-07-10 02:39:01 +08:00
psp_head.py [Fix] fix fast scnn (#606) 2021-07-02 17:58:35 +08:00
segformer_head.py [Feature] Add segformer decode head and related train config (#599) 2021-08-13 13:31:19 +08:00
sep_aspp_head.py [Enhance] Migrate to MMCV DepthwiseSeparableConv (#158) 2020-09-25 19:56:10 +08:00
sep_fcn_head.py [Fix] fix fast scnn (#606) 2021-07-02 17:58:35 +08:00
setr_mla_head.py [Fix] Replace interpolate with resize (#731) 2021-07-28 16:56:22 +08:00
setr_up_head.py [Fix] Replace interpolate with resize (#731) 2021-07-28 16:56:22 +08:00
uper_head.py init commit 2020-07-10 02:39:01 +08:00