[Feature] Update New SegFormer models (open-mmlab#1705)

Author: huajiangjiangLi

configs/segformer/segformer_mit-b1_8xb1-160k_cityscapes-1024x1024.py

@@ -1,7 +1,8 @@
 _base_ = ['./segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b1_20220624-02e5a6a1.pth'  # noqa
 model = dict(
     backbone=dict(
-        init_cfg=dict(type='Pretrained', checkpoint='pretrain/mit_b1.pth'),
+        init_cfg=dict(type='Pretrained', checkpoint=checkpoint),
         embed_dims=64),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b1_8xb2-160k_ade20k-512x512.py

@@ -1,8 +1,10 @@
 _base_ = ['./segformer_mit-b0_8xb2-160k_ade20k-512x512.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b1_20220624-02e5a6a1.pth'  # noqa
+
 # model settings
 model = dict(
-    pretrained='pretrain/mit_b1.pth',
+    pretrained=checkpoint,
     backbone=dict(
         embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[2, 2, 2, 2]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b2_8xb1-160k_cityscapes-1024x1024.py

@@ -1,8 +1,9 @@
 _base_ = ['./segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b2_20220624-66e8bf70.pth'  # noqa
 model = dict(
     backbone=dict(
-        init_cfg=dict(type='Pretrained', checkpoint='pretrain/mit_b2.pth'),
+        init_cfg=dict(type='Pretrained', checkpoint=checkpoint),
         embed_dims=64,
         num_layers=[3, 4, 6, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b2_8xb2-160k_ade20k-512x512.py

@@ -1,8 +1,10 @@
 _base_ = ['./segformer_mit-b0_8xb2-160k_ade20k-512x512.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b2_20220624-66e8bf70.pth'  # noqa
+
 # model settings
 model = dict(
-    pretrained='pretrain/mit_b2.pth',
+    pretrained=checkpoint,
     backbone=dict(
         embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 4, 6, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b3_8xb1-160k_cityscapes-1024x1024.py

@@ -1,8 +1,9 @@
 _base_ = ['./segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b3_20220624-13b1141c.pth'  # noqa
 model = dict(
     backbone=dict(
-        init_cfg=dict(type='Pretrained', checkpoint='pretrain/mit_b3.pth'),
+        init_cfg=dict(type='Pretrained', checkpoint=checkpoint),
         embed_dims=64,
         num_layers=[3, 4, 18, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b3_8xb2-160k_ade20k-512x512.py

@@ -1,8 +1,10 @@
 _base_ = ['./segformer_mit-b0_8xb2-160k_ade20k-512x512.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b3_20220624-13b1141c.pth'  # noqa
+
 # model settings
 model = dict(
-    pretrained='pretrain/mit_b3.pth',
+    pretrained=checkpoint,
     backbone=dict(
         embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 4, 18, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b4_8xb1-160k_cityscapes-1024x1024.py

@@ -1,8 +1,9 @@
 _base_ = ['./segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b4_20220624-d588d980.pth'  # noqa
 model = dict(
     backbone=dict(
-        init_cfg=dict(type='Pretrained', checkpoint='pretrain/mit_b4.pth'),
+        init_cfg=dict(type='Pretrained', checkpoint=checkpoint),
         embed_dims=64,
         num_layers=[3, 8, 27, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b4_8xb2-160k_ade20k-512x512.py

@@ -1,8 +1,10 @@
 _base_ = ['./segformer_mit-b0_8xb2-160k_ade20k-512x512.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b4_20220624-d588d980.pth'  # noqa
+
 # model settings
 model = dict(
-    pretrained='pretrain/mit_b4.pth',
+    pretrained=checkpoint,
     backbone=dict(
         embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 8, 27, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b5_8xb1-160k_cityscapes-1024x1024.py

@@ -1,8 +1,9 @@
 _base_ = ['./segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b5_20220624-658746d9.pthh'  # noqa
 model = dict(
     backbone=dict(
-        init_cfg=dict(type='Pretrained', checkpoint='pretrain/mit_b5.pth'),
+        init_cfg=dict(type='Pretrained', checkpoint=checkpoint),
         embed_dims=64,
         num_layers=[3, 6, 40, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))

configs/segformer/segformer_mit-b5_8xb2-160k_ade20k-512x512.py

@@ -1,8 +1,10 @@
 _base_ = ['./segformer_mit-b0_8xb2-160k_ade20k-512x512.py']
 
+checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/segformer/mit_b5_20220624-658746d9.pthh'  # noqa
+
 # model settings
 model = dict(
-    pretrained='pretrain/mit_b5.pth',
+    pretrained=checkpoint,
     backbone=dict(
         embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 6, 40, 3]),
     decode_head=dict(in_channels=[64, 128, 320, 512]))