mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity

[Config] fix scanner and sar config (#1196)

Browse Source
This commit is contained in:
liukuikun 2022-07-25 11:25:39 +08:00 committed by GitHub
parent 0dc33189e0
commit 2487c0a4a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 58 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
configs/textrecog/robust_scanner/robustscanner_r31_academic.py
View File

@ -5,8 +5,6 @@ _base_ = [
'../../_base_/schedules/schedule_adam_step_5e.py' '../../_base_/schedules/schedule_adam_step_5e.py'
] ]
# dataset settings
train_list = {{_base_.train_list}}
file_client_args = dict(backend='disk') file_client_args = dict(backend='disk')
default_hooks = dict(logger=dict(type='LoggerHook', interval=100)) default_hooks = dict(logger=dict(type='LoggerHook', interval=100))
@ -34,13 +32,41 @@ test_pipeline = [
'instances')) 'instances'))
] ]
# dataset settings
ic11_rec_train = _base_.ic11_rec_train
ic13_rec_train = _base_.ic13_rec_train
ic15_rec_train = _base_.ic15_rec_train
cocov1_rec_train = _base_.cocov1_rec_train
iiit5k_rec_train = _base_.iiit5k_rec_train
st_add_rec_train = _base_.st_add_rec_train
st_rec_train = _base_.st_rec_train
mj_rec_trian = _base_.mj_rec_trian
ic11_rec_train.pipeline = test_pipeline
ic13_rec_train.pipeline = test_pipeline
ic15_rec_train.pipeline = test_pipeline
cocov1_rec_train.pipeline = test_pipeline
iiit5k_rec_train.pipeline = test_pipeline
st_add_rec_train.pipeline = test_pipeline
st_rec_train.pipeline = test_pipeline
mj_rec_trian.pipeline = test_pipeline
repeat_ic11 = dict(type='RepeatDataset', dataset=ic11_rec_train, times=20)
repeat_ic13 = dict(type='RepeatDataset', dataset=ic13_rec_train, times=20)
repeat_ic15 = dict(type='RepeatDataset', dataset=ic15_rec_train, times=20)
repeat_cocov1 = dict(type='RepeatDataset', dataset=cocov1_rec_train, times=20)
repeat_iiit5k = dict(type='RepeatDataset', dataset=iiit5k_rec_train, times=20)
train_dataloader = dict( train_dataloader = dict(
batch_size=64, batch_size=64,
num_workers=8, num_workers=8,
persistent_workers=True, persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True), sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict( dataset=dict(
type='ConcatDataset', datasets=train_list, pipeline=train_pipeline)) type='ConcatDataset',
datasets=[
repeat_ic11, repeat_ic13, repeat_ic15, repeat_cocov1,
repeat_iiit5k, st_add_rec_train, st_rec_train, mj_rec_trian
]))
test_cfg = dict(type='MultiTestLoop') test_cfg = dict(type='MultiTestLoop')
val_cfg = dict(type='MultiValLoop') val_cfg = dict(type='MultiValLoop')

31
configs/textrecog/sar/sar_r31_parallel_decoder_academic.py
View File

@ -7,7 +7,6 @@ _base_ = [
] ]
# dataset settings # dataset settings
train_list = {{_base_.train_list}}
file_client_args = dict(backend='disk') file_client_args = dict(backend='disk')
default_hooks = dict(logger=dict(type='LoggerHook', interval=100)) default_hooks = dict(logger=dict(type='LoggerHook', interval=100))
@ -41,13 +40,41 @@ test_pipeline = [
'instances')) 'instances'))
] ]
# dataset settings
ic11_rec_train = _base_.ic11_rec_train
ic13_rec_train = _base_.ic13_rec_train
ic15_rec_train = _base_.ic15_rec_train
cocov1_rec_train = _base_.cocov1_rec_train
iiit5k_rec_train = _base_.iiit5k_rec_train
st_add_rec_train = _base_.st_add_rec_train
st_rec_train = _base_.st_rec_train
mj_rec_trian = _base_.mj_rec_trian
ic11_rec_train.pipeline = test_pipeline
ic13_rec_train.pipeline = test_pipeline
ic15_rec_train.pipeline = test_pipeline
cocov1_rec_train.pipeline = test_pipeline
iiit5k_rec_train.pipeline = test_pipeline
st_add_rec_train.pipeline = test_pipeline
st_rec_train.pipeline = test_pipeline
mj_rec_trian.pipeline = test_pipeline
repeat_ic11 = dict(type='RepeatDataset', dataset=ic11_rec_train, times=20)
repeat_ic13 = dict(type='RepeatDataset', dataset=ic13_rec_train, times=20)
repeat_ic15 = dict(type='RepeatDataset', dataset=ic15_rec_train, times=20)
repeat_cocov1 = dict(type='RepeatDataset', dataset=cocov1_rec_train, times=20)
repeat_iiit5k = dict(type='RepeatDataset', dataset=iiit5k_rec_train, times=20)
train_dataloader = dict( train_dataloader = dict(
batch_size=64, batch_size=64,
num_workers=8, num_workers=8,
persistent_workers=True, persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True), sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict( dataset=dict(
type='ConcatDataset', datasets=train_list, pipeline=train_pipeline)) type='ConcatDataset',
datasets=[
repeat_ic11, repeat_ic13, repeat_ic15, repeat_cocov1,
repeat_iiit5k, st_add_rec_train, st_rec_train, mj_rec_trian
]))
test_cfg = dict(type='MultiTestLoop') test_cfg = dict(type='MultiTestLoop')
val_cfg = dict(type='MultiValLoop') val_cfg = dict(type='MultiValLoop')