Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Feature] Segformer backbone re-implementation #594

Merged
merged 31 commits into from
Jul 19, 2021
Merged

[Feature] Segformer backbone re-implementation #594

Show file tree
Hide file tree
Changes from 17 commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
b56ceb2
[Feature]Segformer re-implementation
Jun 9, 2021
5a54664
Using act_cfg and norm_cfg to control activation and normalization
Jun 9, 2021
0d60aeb
Split this PR into several little PRs
Jun 10, 2021
c2b0ffd
Fix lint error
Jun 10, 2021
c7e676e
Remove SegFormerHead
Jun 10, 2021
237bc9e
Merge Master
Jun 18, 2021
9edbddd
parameters init refactor
Jun 19, 2021
99d7308
1. Refactor segformer backbone parameters init;
Jun 20, 2021
02b09bf
Merge branch 'master' into segformer
Jun 20, 2021
3373827
Remove rebundant codes
Jun 20, 2021
d967a9f
1. Remove rebundant codes;
Jun 20, 2021
47326de
Refactor the backbone of segformer using mmcv.cnn.bricks.transformer.py
Jun 21, 2021
589af69
Fix some code logic bugs.
Jun 21, 2021
28099de
Add mit_convert.py to match pretrain keys of segformer.
Jun 22, 2021
0f42462
Merge Master
Jun 23, 2021
e3b7157
Merge Master
Jul 6, 2021
a2110ae
Resolve some comments.
Jul 6, 2021
8b7ca03
1. Add some assert to ensure right params;
Jul 9, 2021
754789a
Merge branch 'master' into segformer
Jul 9, 2021
e137df3
Add pe_index assert and fix unit test.
Jul 9, 2021
b4ae0b5
1. Add doc string for MixVisionTransformer;
Jul 12, 2021
38f7bbe
Use hw_shape to pass shape of feature map.
Jul 14, 2021
443438f
1. Fix doc string of MixVisionTransformer;
Jul 14, 2021
1affeb6
Add more unit tests.
Jul 14, 2021
e969496
Add doc string for shape convertion functions.
Jul 14, 2021
582939b
Add some unit tests to improve code coverage.
Jul 14, 2021
e5acfe8
Fix Segformer backbone pretrain weights match bug.
Jul 15, 2021
71df617
resolve the shape convertion functions doc string.
Jul 15, 2021
5ac60d4
Merge branch 'master' of github.com:open-mmlab/mmsegmentation into se…
Jul 15, 2021
38b2db8
Add pad_to_patch_size arg.
Jul 16, 2021
e7039ff
Modify default value of pad_to_patch_size arg.
Jul 19, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion mmseg/models/backbones/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from .cgnet import CGNet
from .fast_scnn import FastSCNN
from .hrnet import HRNet
from .mit import MixVisionTransformer
from .mobilenet_v2 import MobileNetV2
from .mobilenet_v3 import MobileNetV3
from .resnest import ResNeSt
Expand All @@ -13,5 +14,5 @@
__all__ = [
'ResNet', 'ResNetV1c', 'ResNetV1d', 'ResNeXt', 'HRNet', 'FastSCNN',
'ResNeSt', 'MobileNetV2', 'UNet', 'CGNet', 'MobileNetV3',
'VisionTransformer', 'SwinTransformer'
'VisionTransformer', 'SwinTransformer', 'MixVisionTransformer'
]
Loading