| dataset_type = 'OCRDataset' |
|
|
| root = 'tests/data/ocr_toy_dataset' |
| img_prefix = f'{root}/imgs' |
| train_anno_file1 = f'{root}/label.txt' |
|
|
| train1 = dict( |
| type=dataset_type, |
| img_prefix=img_prefix, |
| ann_file=train_anno_file1, |
| loader=dict( |
| type='AnnFileLoader', |
| repeat=100, |
| file_format='txt', |
| file_storage_backend='disk', |
| parser=dict( |
| type='LineStrParser', |
| keys=['filename', 'text'], |
| keys_idx=[0, 1], |
| separator=' ')), |
| pipeline=None, |
| test_mode=False) |
|
|
| train_anno_file2 = f'{root}/label.lmdb' |
| train2 = dict( |
| type=dataset_type, |
| img_prefix=img_prefix, |
| ann_file=train_anno_file2, |
| loader=dict( |
| type='AnnFileLoader', |
| repeat=100, |
| file_format='lmdb', |
| file_storage_backend='disk', |
| parser=dict(type='LineJsonParser', keys=['filename', 'text'])), |
| pipeline=None, |
| test_mode=False) |
|
|
| test_anno_file1 = f'{root}/label.lmdb' |
| test = dict( |
| type=dataset_type, |
| img_prefix=img_prefix, |
| ann_file=test_anno_file1, |
| loader=dict( |
| type='AnnFileLoader', |
| repeat=1, |
| file_format='lmdb', |
| file_storage_backend='disk', |
| parser=dict(type='LineJsonParser', keys=['filename', 'text'])), |
| pipeline=None, |
| test_mode=True) |
|
|
| train_list = [train1, train2] |
|
|
| test_list = [test] |
|
|
