mmocr/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc...

33 lines
831 B
Python
Raw Normal View History

_base_ = [
'_base_dbnetpp_resnet50-dcnv2_fpnc.py',
'../_base_/default_runtime.py',
'../_base_/datasets/synthtext.py',
'../_base_/schedules/schedule_sgd_100k.py',
]
# dataset settings
train_list = [_base_.st_det_train]
test_list = [_base_.st_det_test]
train_dataloader = dict(
batch_size=16,
num_workers=8,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(
type='ConcatDataset',
datasets=train_list,
pipeline=_base_.train_pipeline))
val_dataloader = dict(
batch_size=16,
num_workers=8,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type='ConcatDataset',
datasets=test_list,
pipeline=_base_.test_pipeline))
test_dataloader = val_dataloader