-
Notifications
You must be signed in to change notification settings - Fork 3
/
cocostuff164k.yaml
58 lines (52 loc) · 1.09 KB
/
cocostuff164k.yaml
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
EXP:
ID: cocostuff164k
OUTPUT_DIR: data
DATASET:
NAME: cocostuff164k
ROOT: /media/kazuto1011/Extra/cocostuff/cocostuff-164k
LABELS: ./deeplab-pytorch/data/datasets/cocostuff/labels.txt
N_CLASSES: 182
IGNORE_LABEL: 255
SCALES: [0.5, 0.75, 1.0, 1.25, 1.5]
SPLIT:
TRAIN: train2017
VAL: val2017
TEST:
DATALOADER:
NUM_WORKERS: 0
IMAGE:
MEAN:
R: 122.675
G: 116.669
B: 104.008
SIZE:
BASE: # None
TRAIN: 321
TEST: 512
MODEL:
NAME: DeepLabV2_ResNet101_MSC
N_BLOCKS: [3, 4, 23, 3]
ATROUS_RATES: [6, 12, 18, 24]
INIT_MODEL: deeplab-pytorch/data/models/coco/deeplabv1_resnet101/caffemodel/deeplabv1_resnet101-coco.pth
SOLVER:
BATCH_SIZE:
TRAIN: 5
TEST: 1
ITER_MAX: 100000
ITER_SIZE: 2
ITER_SAVE: 5000
ITER_TB: 100
LR_DECAY: 10
LR: 2.5e-4
MOMENTUM: 0.9
OPTIMIZER: sgd
POLY_POWER: 0.9
WEIGHT_DECAY: 5.0e-4
AVERAGE_LOSS: 20
CRF:
ITER_MAX: 10
POS_W: 3
POS_XY_STD: 1
BI_W: 4
BI_XY_STD: 67
BI_RGB_STD: 3