File size: 842 Bytes
a92ae91 |
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 |
prefix = 'tests/data/ocr_char_ann_toy_dataset/'
train = dict(
type='OCRSegDataset',
img_prefix=f'{prefix}/imgs',
ann_file=f'{prefix}/instances_train.txt',
loader=dict(
type='AnnFileLoader',
repeat=100,
file_format='txt',
parser=dict(
type='LineJsonParser', keys=['file_name', 'annotations', 'text'])),
pipeline=None,
test_mode=True)
test = dict(
type='OCRDataset',
img_prefix=f'{prefix}/imgs',
ann_file=f'{prefix}/instances_test.txt',
loader=dict(
type='AnnFileLoader',
repeat=1,
file_format='txt',
parser=dict(
type='LineStrParser',
keys=['filename', 'text'],
keys_idx=[0, 1],
separator=' ')),
pipeline=None,
test_mode=True)
train_list = [train]
test_list = [test]
|