Skip to content

Commit f4d9800

Browse files
update pascalcontext configs
1 parent 607dd18 commit f4d9800

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

ssseg/configs/deeplabv3/deeplabv3_resnet50os8_pascalcontext.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,9 @@
2626
'in_channels': 2048, 'feats_channels': 512, 'dilations': [1, 12, 24, 36], 'dropout': 0.1,
2727
}
2828
SEGMENTOR_CFG['inference'] = {
29-
'mode': 'slide',
30-
'opts': {'cropsize': (480, 480), 'stride': (320, 320)},
31-
'tricks': {
32-
'multiscale': [1], 'flip': False, 'use_probs_before_resize': True,
33-
}
29+
'forward': {'mode': 'slide', 'cropsize': (480, 480), 'stride': (320, 320)},
30+
'tta': {'multiscale': [1], 'flip': False, 'use_probs_before_resize': True},
31+
'evaluate': {'metric_list': ['iou', 'miou']},
3432
}
3533
SEGMENTOR_CFG['work_dir'] = os.path.split(__file__)[-1].split('.')[0]
3634
SEGMENTOR_CFG['logger_handle_cfg']['logfilepath'] = os.path.join(SEGMENTOR_CFG['work_dir'], f"{os.path.split(__file__)[-1].split('.')[0]}.log")

ssseg/configs/deeplabv3/deeplabv3_resnet50os8_pascalcontext59.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,9 @@
2626
'in_channels': 2048, 'feats_channels': 512, 'dilations': [1, 12, 24, 36], 'dropout': 0.1,
2727
}
2828
SEGMENTOR_CFG['inference'] = {
29-
'mode': 'slide',
30-
'opts': {'cropsize': (480, 480), 'stride': (320, 320)},
31-
'tricks': {
32-
'multiscale': [1], 'flip': False, 'use_probs_before_resize': True,
33-
}
29+
'forward': {'mode': 'slide', 'cropsize': (480, 480), 'stride': (320, 320)},
30+
'tta': {'multiscale': [1], 'flip': False, 'use_probs_before_resize': True},
31+
'evaluate': {'metric_list': ['iou', 'miou']},
3432
}
3533
SEGMENTOR_CFG['work_dir'] = os.path.split(__file__)[-1].split('.')[0]
3634
SEGMENTOR_CFG['logger_handle_cfg']['logfilepath'] = os.path.join(SEGMENTOR_CFG['work_dir'], f"{os.path.split(__file__)[-1].split('.')[0]}.log")

0 commit comments

Comments
 (0)