-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcfg_voc20.py
32 lines (28 loc) · 951 Bytes
/
cfg_voc20.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# VOC20 config file that define test pipeline
# Please refer to here https://github.com/wangf3014/SCLIP/blob/main/configs/cfg_voc20.py
_base_ = './base_config.py'
# model settings
model = dict(
name_path='./training/seg_configs/cls_voc20.txt'
)
# dataset settings
dataset_type = 'PascalVOC20Dataset'
data_root = '/mmsegmentation_datasets/data/VOCdevkit/VOC2012'
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='Resize', scale=(2048, 336), keep_ratio=True),
dict(type='LoadAnnotations'),
dict(type='PackSegInputs')
]
test_dataloader = dict(
batch_size=1,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
data_prefix=dict(
img_path='JPEGImages', seg_map_path='SegmentationClass'),
ann_file='ImageSets/Segmentation/val.txt',
pipeline=test_pipeline))