Skip to content

Commit 733ad9e

Browse files
[Fix] Fix Segmenter-l readme (open-mmlab#1695)
* [Fix] Fix Segmenter-l readme * fix
1 parent e8dfa3f commit 733ad9e

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

configs/segmenter/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ In our default setting, pretrained models and their corresponding [ViT-AugReg](h
7171
| Segmenter Linear | ViT-S_16 | 512x512 | 160000 | 1.78 | 28.07 | 45.75 | 46.82 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/segmenter/segmenter_vit-s_linear_8x1_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-s_linear_8x1_512x512_160k_ade20k/segmenter_vit-s_linear_8x1_512x512_160k_ade20k_20220105_151713-39658c46.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-s_linear_8x1_512x512_160k_ade20k/segmenter_vit-s_linear_8x1_512x512_160k_ade20k_20220105_151713.log.json) |
7272
| Segmenter Mask | ViT-S_16 | 512x512 | 160000 | 2.03 | 24.80 | 46.19 | 47.85 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/segmenter/segmenter_vit-s_mask_8x1_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-s_mask_8x1_512x512_160k_ade20k/segmenter_vit-s_mask_8x1_512x512_160k_ade20k_20220105_151706-511bb103.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-s_mask_8x1_512x512_160k_ade20k/segmenter_vit-s_mask_8x1_512x512_160k_ade20k_20220105_151706.log.json) |
7373
| Segmenter Mask | ViT-B_16 | 512x512 | 160000 | 4.20 | 13.20 | 49.60 | 51.07 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/segmenter/segmenter_vit-b_mask_8x1_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-b_mask_8x1_512x512_160k_ade20k/segmenter_vit-b_mask_8x1_512x512_160k_ade20k_20220105_151706-bc533b08.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-b_mask_8x1_512x512_160k_ade20k/segmenter_vit-b_mask_8x1_512x512_160k_ade20k_20220105_151706.log.json) |
74-
| Segmenter Mask | ViT-L_16 | 640x640 | 160000 | 16.56 | 2.62 | 52.16 | 53.65 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k/segmenter_vit-l_mask_8x1_512x512_160k_ade20k_20220105_162750-7ef345be.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k/segmenter_vit-l_mask_8x1_512x512_160k_ade20k_20220105_162750.log.json) |
74+
| Segmenter Mask | ViT-L_16 | 640x640 | 160000 | 16.99 | 3.03 | 51.65 | 53.58 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k/segmenter_vit-l_mask_8x1_640x640_160k_ade20k_20220614_024513-4783a347.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k/segmenter_vit-l_mask_8x1_640x640_160k_ade20k_20220614_024513.log.json) |
7575

7676
Note:
7777

configs/segmenter/segmenter.yml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,25 +101,25 @@ Models:
101101
mIoU(ms+flip): 51.07
102102
Config: configs/segmenter/segmenter_vit-b_mask_8x1_512x512_160k_ade20k.py
103103
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-b_mask_8x1_512x512_160k_ade20k/segmenter_vit-b_mask_8x1_512x512_160k_ade20k_20220105_151706-bc533b08.pth
104-
- Name: segmenter_vit-l_mask_8x1_512x512_160k_ade20k
104+
- Name: segmenter_vit-l_mask_8x1_640x640_160k_ade20k
105105
In Collection: Segmenter
106106
Metadata:
107107
backbone: ViT-L_16
108108
crop size: (640,640)
109109
lr schd: 160000
110110
inference time (ms/im):
111-
- value: 381.68
111+
- value: 330.03
112112
hardware: V100
113113
backend: PyTorch
114114
batch size: 1
115115
mode: FP32
116116
resolution: (640,640)
117-
Training Memory (GB): 16.56
117+
Training Memory (GB): 16.99
118118
Results:
119119
- Task: Semantic Segmentation
120120
Dataset: ADE20K
121121
Metrics:
122-
mIoU: 52.16
123-
mIoU(ms+flip): 53.65
124-
Config: configs/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k.py
125-
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k/segmenter_vit-l_mask_8x1_512x512_160k_ade20k_20220105_162750-7ef345be.pth
122+
mIoU: 51.65
123+
mIoU(ms+flip): 53.58
124+
Config: configs/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k.py
125+
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k/segmenter_vit-l_mask_8x1_640x640_160k_ade20k_20220614_024513-4783a347.pth

configs/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k.py renamed to configs/segmenter/segmenter_vit-l_mask_8x1_640x640_160k_ade20k.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
_base_ = [
22
'../_base_/models/segmenter_vit-b16_mask.py',
3-
'../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
3+
'../_base_/datasets/ade20k_640x640.py', '../_base_/default_runtime.py',
44
'../_base_/schedules/schedule_160k.py'
55
]
66
checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segmenter/vit_large_p16_384_20220308-d4efb41d.pth' # noqa
@@ -29,7 +29,7 @@
2929
train_pipeline = [
3030
dict(type='LoadImageFromFile'),
3131
dict(type='LoadAnnotations', reduce_zero_label=True),
32-
dict(type='Resize', img_scale=(2048, 640), ratio_range=(0.5, 2.0)),
32+
dict(type='Resize', img_scale=(2560, 640), ratio_range=(0.5, 2.0)),
3333
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
3434
dict(type='RandomFlip', prob=0.5),
3535
dict(type='PhotoMetricDistortion'),
@@ -42,7 +42,7 @@
4242
dict(type='LoadImageFromFile'),
4343
dict(
4444
type='MultiScaleFlipAug',
45-
img_scale=(2048, 640),
45+
img_scale=(2560, 640),
4646
# img_ratios=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75],
4747
flip=False,
4848
transforms=[

0 commit comments

Comments
 (0)