-
Notifications
You must be signed in to change notification settings - Fork 4
/
segmentation_base.yaml
152 lines (122 loc) · 2.43 KB
/
segmentation_base.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
_BASE_: "../Base-RCNN-FPN.yaml"
MODEL:
META_ARCHITECTURE: "SemanticSegmentor"
BACKBONE:
FREEZE_AT: 0
RESUME: False
PIXEL_MEAN: [123.675, 116.280, 103.530]
PIXEL_STD: [58.395, 57.120, 57.375]
DATASETS:
TRAIN: ("train",)
TEST: ("val",)
DATALOADER:
NUM_WORKERS: 16
FILTER_EMPTY_ANNOTATIONS: False
PREPROCESS:
OVERWRITE: False
DISABLE_CHECK: False
RESIZE:
RESIZE_MODE: "shortest_edge"
RESIZE_SAMPLING: "choice"
MIN_SIZE: [1024]
MAX_SIZE: 2048
SCALING: 0.5
REGION:
REGIONS:
[
"marginalia",
"page-number",
"resolution",
"date",
"index",
"attendance",
"Resumption",
"resumption",
"Insertion",
"insertion",
]
MERGE_REGIONS: ["resolution:Resumption,resumption,Insertion,insertion"]
REGION_TYPE: []
BASELINE:
LINE_WIDTH: 5
INPUT:
RESIZE_MODE: "shortest_edge"
MIN_SIZE_TRAIN_SAMPLING: choice
MIN_SIZE_TRAIN: (1024,)
MAX_SIZE_TRAIN: 2048
SCALING_TRAIN: 1.0
# FIXME Have the Min size adjustable
MIN_SIZE_TEST: 1024
MAX_SIZE_TEST: 2048
SCALING_TEST: -1.
FORMAT: RGB
GRAYSCALE:
PROBABILITY: 0.
ADAPTIVE_THRESHOLDING:
PROBABILITY: 0.
INVERT:
PROBABILITY: 0.
JPEG_COMPRESSION:
PROBABILITY: 0.
MIN_QUALITY: 50
MAX_QUALITY: 100
NOISE:
PROBABILITY: 0.
MIN_STD: 10
MAX_STD: 32
BRIGHTNESS:
PROBABILITY: 0.
MIN_INTENSITY: 0.5
MAX_INTENSITY: 1.5
CONTRAST:
PROBABILITY: 0.
MIN_INTENSITY: 0.5
MAX_INTENSITY: 1.5
SATURATION:
PROBABILITY: 0.
MIN_INTENSITY: 0.5
MAX_INTENSITY: 1.5
HUE:
PROBABILITY: 0.
MIN_DELTA: -0.25
MAX_DELTA: 0.25
GAUSSIAN_FILTER:
PROBABILITY: 0.
MIN_SIGMA: 0.5
MAX_SIGMA: 1.5
HORIZONTAL_FLIP:
PROBABILITY: 0.
VERTICAL_FLIP:
PROBABILITY: 0.
ELASTIC_DEFORMATION:
PROBABILITY: 0.
ALPHA: 0.1
SIGMA: 0.01
AFFINE:
PROBABILITY: 0.
TRANSLATION:
PROBABILITY: 0.5
STANDARD_DEVIATION: 0.02
ROTATION:
PROBABILITY: 0.5
KAPPA: 30.
SHEAR:
PROBABILITY: 0.5
KAPPA: 20.
SCALE:
PROBABILITY: 0.5
STANDARD_DEVIATION: 0.12
ORIENTATION:
PROBABILITY: 0.
PERCENTAGES: (0.25, 0.25, 0.25, 0.25)
TEST:
EVAL_PERIOD: 10000
SOLVER:
IMS_PER_BATCH: 8
CHECKPOINT_PERIOD: 25000
BASE_LR: 0.0002
GAMMA: 0.1
STEPS: () #(80000, 120000, 160000)
MAX_ITER: 250000
OUTPUT_DIR: ./output
SEED: 42