Skip to content

Commit f3cf96b

Browse files
authored
[Bug fix] Fixed ADE20k test (open-mmlab#359)
* [Bug fix] Fixed ADE20k test * fixed ade cfg
1 parent 4423d32 commit f3cf96b

17 files changed

+9
-25
lines changed

configs/apcnet/apcnet_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/apcnet/apcnet_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/deeplabv3/deeplabv3_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/deeplabv3/deeplabv3_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/dmnet/dmnet_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/dmnet/dmnet_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/fcn/fcn_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/fcn/fcn_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/psanet/psanet_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,3 @@
55
model = dict(
66
decode_head=dict(mask_size=(66, 66), num_classes=150),
77
auxiliary_head=dict(num_classes=150))
8-
test_cfg = dict(mode='whole')

configs/psanet/psanet_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,3 @@
55
model = dict(
66
decode_head=dict(mask_size=(66, 66), num_classes=150),
77
auxiliary_head=dict(num_classes=150))
8-
test_cfg = dict(mode='whole')

configs/pspnet/pspnet_r50-d8_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/pspnet/pspnet_r50-d8_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/upernet/upernet_r50_512x512_160k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

configs/upernet/upernet_r50_512x512_80k_ade20k.py

-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,3 @@
44
]
55
model = dict(
66
decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7-
test_cfg = dict(mode='whole')

mmseg/core/evaluation/metrics.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,15 @@ def intersect_and_union(pred_label,
3434

3535
if isinstance(label, str):
3636
label = mmcv.imread(label, flag='unchanged', backend='pillow')
37-
# modify if custom classes
38-
if label_map is not None:
39-
for old_id, new_id in label_map.items():
40-
label[label == old_id] = new_id
41-
if reduce_zero_label:
42-
# avoid using underflow conversion
43-
label[label == 0] = 255
44-
label = label - 1
45-
label[label == 254] = 255
37+
# modify if custom classes
38+
if label_map is not None:
39+
for old_id, new_id in label_map.items():
40+
label[label == old_id] = new_id
41+
if reduce_zero_label:
42+
# avoid using underflow conversion
43+
label[label == 0] = 255
44+
label = label - 1
45+
label[label == 254] = 255
4646

4747
mask = (label != ignore_index)
4848
pred_label = pred_label[mask]

0 commit comments

Comments
 (0)