-
Notifications
You must be signed in to change notification settings - Fork 747
/
sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
69 lines (63 loc) · 2.17 KB
/
sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.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
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
_base_ = [
'../_base_/datasets/mjsynth.py',
'../_base_/datasets/synthtext.py',
'../_base_/datasets/synthtext_add.py',
'../_base_/datasets/coco_text_v1.py',
'../_base_/datasets/cute80.py',
'../_base_/datasets/iiit5k.py',
'../_base_/datasets/svt.py',
'../_base_/datasets/svtp.py',
'../_base_/datasets/icdar2011.py',
'../_base_/datasets/icdar2013.py',
'../_base_/datasets/icdar2015.py',
'../_base_/default_runtime.py',
'../_base_/schedules/schedule_adam_step_5e.py',
'_base_sar_resnet31_parallel-decoder.py',
]
default_hooks = dict(logger=dict(type='LoggerHook', interval=100))
# dataset settings
train_list = [
_base_.icdar2011_textrecog_train, _base_.icdar2013_textrecog_train,
_base_.icdar2015_textrecog_train, _base_.cocotextv1_textrecog_train,
_base_.iiit5k_textrecog_train, _base_.mjsynth_sub_textrecog_train,
_base_.synthtext_sub_textrecog_train, _base_.synthtext_add_textrecog_train
]
test_list = [
_base_.cute80_textrecog_test, _base_.iiit5k_textrecog_test,
_base_.svt_textrecog_test, _base_.svtp_textrecog_test,
_base_.icdar2013_textrecog_test, _base_.icdar2015_textrecog_test
]
train_list = [
dict(
type='RepeatDataset',
dataset=dict(
type='ConcatDataset',
datasets=train_list[:5],
pipeline=_base_.train_pipeline),
times=20),
dict(
type='ConcatDataset',
datasets=train_list[5:],
pipeline=_base_.train_pipeline),
]
train_dataloader = dict(
batch_size=64 * 6,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='ConcatDataset', datasets=train_list, verify_meta=False))
test_dataloader = dict(
batch_size=1,
num_workers=4,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type='ConcatDataset',
datasets=test_list,
pipeline=_base_.test_pipeline))
val_dataloader = test_dataloader
val_evaluator = dict(
dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
test_evaluator = val_evaluator
auto_scale_lr = dict(base_batch_size=64 * 48)