|
2 | 2 | # Inference Speed is tested on NVIDIA V100 |
3 | 3 | hrnet = [ |
4 | 4 | dict( |
5 | | - config='configs/hrnet/fcn_hr18s_512x512_160k_ade20k.py', |
| 5 | + config='configs/hrnet/fcn_hr18s_4xb4-160k_ade20k-512x512.py', |
6 | 6 | checkpoint='fcn_hr18s_512x512_160k_ade20k_20200614_214413-870f65ac.pth', # noqa |
7 | 7 | eval='mIoU', |
8 | 8 | metric=dict(mIoU=33.0), |
9 | 9 | ), |
10 | 10 | dict( |
11 | | - config='configs/hrnet/fcn_hr18s_512x1024_160k_cityscapes.py', |
| 11 | + config='configs/hrnet/fcn_hr18s_4xb2-160k_cityscapes-512x1024.py', |
12 | 12 | checkpoint='fcn_hr18s_512x1024_160k_cityscapes_20200602_190901-4a0797ea.pth', # noqa |
13 | 13 | eval='mIoU', |
14 | 14 | metric=dict(mIoU=76.31), |
15 | 15 | ), |
16 | 16 | dict( |
17 | | - config='configs/hrnet/fcn_hr48_512x512_160k_ade20k.py', |
| 17 | + config='configs/hrnet/fcn_hr48_4xb4-160k_ade20k-512x512.py', |
18 | 18 | checkpoint='fcn_hr48_512x512_160k_ade20k_20200614_214407-a52fc02c.pth', |
19 | 19 | eval='mIoU', |
20 | 20 | metric=dict(mIoU=42.02), |
21 | 21 | ), |
22 | 22 | dict( |
23 | | - config='configs/hrnet/fcn_hr48_512x1024_160k_cityscapes.py', |
| 23 | + config='configs/hrnet/fcn_hr48_4xb2-160k_cityscapes-512x1024.py', |
24 | 24 | checkpoint='fcn_hr48_512x1024_160k_cityscapes_20200602_190946-59b7973e.pth', # noqa |
25 | 25 | eval='mIoU', |
26 | 26 | metric=dict(mIoU=80.65), |
27 | 27 | ), |
28 | 28 | ] |
29 | 29 | pspnet = [ |
30 | 30 | dict( |
31 | | - config='configs/pspnet/pspnet_r50-d8_512x1024_80k_cityscapes.py', |
| 31 | + config='configs/pspnet/pspnet_r50-d8_4xb2-80k_cityscapes-512x1024.py', |
32 | 32 | checkpoint='pspnet_r50-d8_512x1024_80k_cityscapes_20200606_112131-2376f12b.pth', # noqa |
33 | 33 | eval='mIoU', |
34 | 34 | metric=dict(mIoU=78.55), |
35 | 35 | ), |
36 | 36 | dict( |
37 | | - config='configs/pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py', |
| 37 | + config='configs/pspnet/pspnet_r101-d8_4xb2-80k_cityscapes-512x1024.py', |
38 | 38 | checkpoint='pspnet_r101-d8_512x1024_80k_cityscapes_20200606_112211-e1e1100f.pth', # noqa |
39 | 39 | eval='mIoU', |
40 | 40 | metric=dict(mIoU=79.76), |
41 | 41 | ), |
42 | 42 | dict( |
43 | | - config='configs/pspnet/pspnet_r101-d8_512x512_160k_ade20k.py', |
| 43 | + config='configs/pspnet/pspnet_r101-d8_4xb4-160k_ade20k-512x512.py', |
44 | 44 | checkpoint='pspnet_r101-d8_512x512_160k_ade20k_20200615_100650-967c316f.pth', # noqa |
45 | 45 | eval='mIoU', |
46 | 46 | metric=dict(mIoU=44.39), |
47 | 47 | ), |
48 | 48 | dict( |
49 | | - config='configs/pspnet/pspnet_r50-d8_512x512_160k_ade20k.py', |
| 49 | + config='configs/pspnet/pspnet_r50-d8_4xb4-160k_ade20k-512x512.py', |
50 | 50 | checkpoint='pspnet_r50-d8_512x512_160k_ade20k_20200615_184358-1890b0bd.pth', # noqa |
51 | 51 | eval='mIoU', |
52 | 52 | metric=dict(mIoU=42.48), |
53 | 53 | ), |
54 | 54 | ] |
55 | 55 | resnest = [ |
56 | 56 | dict( |
57 | | - config='configs/resnest/pspnet_s101-d8_512x512_160k_ade20k.py', |
| 57 | + config='configs/resnest/resnest_s101-d8_pspnet_4xb4-160k_ade20k-512x512.py', # noqa |
58 | 58 | checkpoint='pspnet_s101-d8_512x512_160k_ade20k_20200807_145416-a6daa92a.pth', # noqa |
59 | 59 | eval='mIoU', |
60 | 60 | metric=dict(mIoU=45.44), |
61 | 61 | ), |
62 | 62 | dict( |
63 | | - config='configs/resnest/pspnet_s101-d8_512x1024_80k_cityscapes.py', |
| 63 | + config='configs/resnest/resnest_s101-d8_pspnet_4xb2-80k_cityscapes512x1024.py', # noqa |
64 | 64 | checkpoint='pspnet_s101-d8_512x1024_80k_cityscapes_20200807_140631-c75f3b99.pth', # noqa |
65 | 65 | eval='mIoU', |
66 | 66 | metric=dict(mIoU=78.57), |
67 | 67 | ), |
68 | 68 | ] |
69 | 69 | fastscnn = [ |
70 | 70 | dict( |
71 | | - config='configs/fastscnn/fast_scnn_lr0.12_8x4_160k_cityscapes.py', |
| 71 | + config='configs/fastscnn/fast_scnn_8xb4-160k_cityscapes-512x1024.py', |
72 | 72 | checkpoint='fast_scnn_8x4_160k_lr0.12_cityscapes-0cec9937.pth', |
73 | 73 | eval='mIoU', |
74 | 74 | metric=dict(mIoU=70.96), |
75 | 75 | ) |
76 | 76 | ] |
77 | 77 | deeplabv3plus = [ |
78 | 78 | dict( |
79 | | - config='configs/deeplabv3plus/deeplabv3plus_r101-d8_769x769_80k_cityscapes.py', # noqa |
| 79 | + config='configs/deeplabv3plus/deeplabv3plus_r101-d8_4xb2-80k_cityscapes-769x769.py', # noqa |
80 | 80 | checkpoint='deeplabv3plus_r101-d8_769x769_80k_cityscapes_20200607_000405-a7573d20.pth', # noqa |
81 | 81 | eval='mIoU', |
82 | 82 | metric=dict(mIoU=80.98), |
83 | 83 | ), |
84 | 84 | dict( |
85 | | - config='configs/deeplabv3plus/deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py', # noqa |
| 85 | + config='configs/deeplabv3plus/deeplabv3plus_r101-d8_4xb2-80k_cityscapes-512x1024.py', # noqa |
86 | 86 | checkpoint='deeplabv3plus_r101-d8_512x1024_80k_cityscapes_20200606_114143-068fcfe9.pth', # noqa |
87 | 87 | eval='mIoU', |
88 | 88 | metric=dict(mIoU=80.97), |
89 | 89 | ), |
90 | 90 | dict( |
91 | | - config='configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py', # noqa |
| 91 | + config='configs/deeplabv3plus/deeplabv3plus_r50-d8_4xb2-80k_cityscapes-512x1024.py', # noqa |
92 | 92 | checkpoint='deeplabv3plus_r50-d8_512x1024_80k_cityscapes_20200606_114049-f9fb496d.pth', # noqa |
93 | 93 | eval='mIoU', |
94 | 94 | metric=dict(mIoU=80.09), |
95 | 95 | ), |
96 | 96 | dict( |
97 | | - config='configs/deeplabv3plus/deeplabv3plus_r50-d8_769x769_80k_cityscapes.py', # noqa |
| 97 | + config='configs/deeplabv3plus/deeplabv3plus_r50-d8_4xb2-80k_cityscapes-769x769.py', # noqa |
98 | 98 | checkpoint='deeplabv3plus_r50-d8_769x769_80k_cityscapes_20200606_210233-0e9dfdc4.pth', # noqa |
99 | 99 | eval='mIoU', |
100 | 100 | metric=dict(mIoU=79.83), |
101 | 101 | ), |
102 | 102 | ] |
103 | 103 | vit = [ |
104 | 104 | dict( |
105 | | - config='configs/vit/upernet_vit-b16_ln_mln_512x512_160k_ade20k.py', |
| 105 | + config='configs/vit/vit_vit-b16-ln_mln_upernet_8xb2-160k_ade20k-512x512.py', # noqa |
106 | 106 | checkpoint='upernet_vit-b16_ln_mln_512x512_160k_ade20k-f444c077.pth', |
107 | 107 | eval='mIoU', |
108 | 108 | metric=dict(mIoU=47.73), |
109 | 109 | ), |
110 | 110 | dict( |
111 | | - config='configs/vit/upernet_deit-s16_ln_mln_512x512_160k_ade20k.py', |
| 111 | + config='configs/vit/vit_deit-s16-ln_mln_upernet_512x512_160k_ade20k-512x512.py', # noqa |
112 | 112 | checkpoint='upernet_deit-s16_ln_mln_512x512_160k_ade20k-c0cd652f.pth', |
113 | 113 | eval='mIoU', |
114 | 114 | metric=dict(mIoU=43.52), |
115 | 115 | ), |
116 | 116 | ] |
117 | 117 | fp16 = [ |
118 | 118 | dict( |
119 | | - config='configs/deeplabv3plus/deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py', # noqa |
| 119 | + config='configs/deeplabv3plus/deeplabv3plus_r101-d8_4xb2-amp-80k_cityscapes-512x1024.py', # noqa |
120 | 120 | checkpoint='deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes_20200717_230920-f1104f4b.pth', # noqa |
121 | 121 | eval='mIoU', |
122 | 122 | metric=dict(mIoU=80.46), |
123 | 123 | ) |
124 | 124 | ] |
125 | 125 | swin = [ |
126 | 126 | dict( |
127 | | - config='configs/swin/upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py', # noqa |
| 127 | + config='configs/swin/swin-tiny-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py', # noqa |
128 | 128 | checkpoint='upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K_20210531_112542-e380ad3e.pth', # noqa |
129 | 129 | eval='mIoU', |
130 | 130 | metric=dict(mIoU=44.41), |
|
0 commit comments