Skip to content

Commit dba7595

Browse files
committed
added own dataset config
1 parent 6c42485 commit dba7595

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
_base_ = [
2+
'../_base_/models/deeplabv3plus_r50-d8.py',
3+
'../_base_/datasets/pascal_mos.py', '../_base_/default_runtime.py',
4+
'../_base_/schedules/schedule_40k.py'
5+
]
6+
crop_size = (480, 480)
7+
data_preprocessor = dict(size=crop_size)
8+
model = dict(
9+
data_preprocessor=data_preprocessor,
10+
decode_head=dict(num_classes=3),
11+
auxiliary_head=dict(num_classes=3),
12+
test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
13+
optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
14+
optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer)

0 commit comments

Comments
 (0)