Skip to content

Commit 032ad84

Browse files
authored
[Fix] Config name in ut (#1978)
* [Fix] Config name in ut * lint
1 parent c5bcf99 commit 032ad84

File tree

2 files changed

+23
-20
lines changed

2 files changed

+23
-20
lines changed

tests/test_models/test_data_preprocessor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
11
# Copyright (c) OpenMMLab. All rights reserved.
22
from unittest import TestCase
33

4-
from mmseg.models import SegDataPreProcessor
5-
64
import torch
75
from mmengine.structures import PixelData
86

7+
from mmseg.models import SegDataPreProcessor
98
from mmseg.structures import SegDataSample
109

1110

tests/test_models/test_forward.py

Lines changed: 22 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -104,90 +104,94 @@ def _get_segmentor_cfg(fname):
104104

105105
def test_pspnet_forward():
106106
_test_encoder_decoder_forward(
107-
'pspnet/pspnet_r18-d8_512x1024_80k_cityscapes.py')
107+
'pspnet/pspnet_r18-d8_4xb2-80k_cityscapes-512x1024.py')
108108

109109

110110
def test_fcn_forward():
111-
_test_encoder_decoder_forward('fcn/fcn_r18-d8_512x1024_80k_cityscapes.py')
111+
_test_encoder_decoder_forward(
112+
'fcn/fcn_r18-d8_4xb2-80k_cityscapes-512x1024.py')
112113

113114

114115
def test_deeplabv3_forward():
115116
_test_encoder_decoder_forward(
116-
'deeplabv3/deeplabv3_r18-d8_512x1024_80k_cityscapes.py')
117+
'deeplabv3/deeplabv3_r18-d8_4xb2-80k_cityscapes-512x1024.py')
117118

118119

119120
def test_deeplabv3plus_forward():
120121
_test_encoder_decoder_forward(
121-
'deeplabv3plus/deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py')
122+
'deeplabv3plus/deeplabv3plus_r18-d8_4xb2-80k_cityscapes-512x1024.py')
122123

123124

124125
def test_gcnet_forward():
125126
_test_encoder_decoder_forward(
126-
'gcnet/gcnet_r50-d8_512x1024_40k_cityscapes.py')
127+
'gcnet/gcnet_r50-d8_4xb2-40k_cityscapes-512x1024.py')
127128

128129

129130
def test_ann_forward():
130-
_test_encoder_decoder_forward('ann/ann_r50-d8_512x1024_40k_cityscapes.py')
131+
_test_encoder_decoder_forward(
132+
'ann/ann_r50-d8_4xb2-40k_cityscapes-512x1024.py')
131133

132134

133135
def test_ccnet_forward():
134136
if not torch.cuda.is_available():
135137
pytest.skip('CCNet requires CUDA')
136138
_test_encoder_decoder_forward(
137-
'ccnet/ccnet_r50-d8_512x1024_40k_cityscapes.py')
139+
'ccnet/ccnet_r50-d8_4xb2-40k_cityscapes-512x1024.py')
138140

139141

140142
def test_danet_forward():
141143
_test_encoder_decoder_forward(
142-
'danet/danet_r50-d8_512x1024_40k_cityscapes.py')
144+
'danet/danet_r50-d8_4xb2-40k_cityscapes-512x1024.py')
143145

144146

145147
def test_nonlocal_net_forward():
146148
_test_encoder_decoder_forward(
147-
'nonlocal_net/nonlocal_r50-d8_512x1024_40k_cityscapes.py')
149+
'nonlocal_net/nonlocal_r50-d8_4xb2-40k_cityscapes-512x1024.py')
148150

149151

150152
def test_upernet_forward():
151153
_test_encoder_decoder_forward(
152-
'upernet/upernet_r50_512x1024_40k_cityscapes.py')
154+
'upernet/upernet_r50_4xb2-40k_cityscapes-512x1024.py')
153155

154156

155157
def test_hrnet_forward():
156-
_test_encoder_decoder_forward('hrnet/fcn_hr18s_512x1024_40k_cityscapes.py')
158+
_test_encoder_decoder_forward(
159+
'hrnet/fcn_hr18s_4xb2-40k_cityscapes-512x1024.py')
157160

158161

159162
def test_ocrnet_forward():
160163
_test_encoder_decoder_forward(
161-
'ocrnet/ocrnet_hr18s_512x1024_40k_cityscapes.py')
164+
'ocrnet/ocrnet_hr18s_4xb2-40k_cityscapes-512x1024.py')
162165

163166

164167
def test_psanet_forward():
165168
_test_encoder_decoder_forward(
166-
'psanet/psanet_r50-d8_512x1024_40k_cityscapes.py')
169+
'psanet/psanet_r50-d8_4xb2-40k_cityscapes-512x1024.py')
167170

168171

169172
def test_sem_fpn_forward():
170-
_test_encoder_decoder_forward('sem_fpn/fpn_r50_512x1024_80k_cityscapes.py')
173+
_test_encoder_decoder_forward(
174+
'sem_fpn/fpn_r50_4xb2-80k_cityscapes-512x1024.py')
171175

172176

173177
def test_mobilenet_v2_forward():
174178
_test_encoder_decoder_forward(
175-
'mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py')
179+
'mobilenet_v2/mobilenet-v2-d8_pspnet_4xb2-80k_cityscapes-512x1024.py')
176180

177181

178182
def test_dnlnet_forward():
179183
_test_encoder_decoder_forward(
180-
'dnlnet/dnl_r50-d8_512x1024_40k_cityscapes.py')
184+
'dnlnet/dnl_r50-d8_4xb2-40k_cityscapes-512x1024.py')
181185

182186

183187
def test_emanet_forward():
184188
_test_encoder_decoder_forward(
185-
'emanet/emanet_r50-d8_512x1024_80k_cityscapes.py')
189+
'emanet/emanet_r50-d8_4xb2-80k_cityscapes-512x1024.py')
186190

187191

188192
def test_isanet_forward():
189193
_test_encoder_decoder_forward(
190-
'isanet/isanet_r50-d8_512x1024_40k_cityscapes.py')
194+
'isanet/isanet_r50-d8_4xb2-40k_cityscapes-512x1024.py')
191195

192196

193197
def get_world_size(process_group):

0 commit comments

Comments
 (0)